推荐一款用于高性能 AI 研究轻量级的 PyTorch 框架：PyTorch Lightning。

该框架主要特性：

1. 可以在 GPU，TPU，CPU上甚至在16位精度下运行，无需更改模型；

2. 研究代码与工程代码解耦，使代码可读性更高；

3. 集成现有流行的日志记录/可视化框架，可开箱即用；

4. 拥有更快的训练模型速度。

GitHub： 网页链接

https://weibo.com/5722964389/JjwBiA5qm