apple-ml - Apple Silicon 机器学习代码模式
为 Apple Silicon M 系列硬件编写 PyTorch 和 MLX 代码，无需 CUDA
标签
更新于: 2026-03-25能力
典型输入
典型输出
该技能可以做什么
- 选择设备
- 设置内存限制
- 监控内存
- 同步设备
- 设置回退
- 保存检查点
- 调整批次大小
- 调度学习率
- 获取内存
- 设置缓存限制
- 清除缓存
- 求值
- 裁剪梯度
输入
- PyTorch 代码
- MLX 代码
- 系统内存
- 训练超参数
- 模型权重
输出
- 内存状态
- 设备选择
- 检查点文件
- 求值结果
要求
- Apple Silicon M 系列机器
- PyTorch
- MLX