jittor

收录理由

如果你希望尝试一种以即时编译为核心的深度学习框架，而不是常见的类似 PyTorch 的急切执行循环，Jittor 值得关注。它的框架和元算子会在运行时动态编译，因此算子编译器和调优器能够针对你的模型生成定制代码，而不是提供通用的内核。前端采用纯 Python，支持动态图执行和 Module API，从 PyTorch 迁移过来的用户会感到熟悉，而后端则基于 CUDA 和 C++ 以保证性能。除了训练之外，该项目还提供了检测、分割、生成、可微渲染和强化学习等模型库，既适合复现已发表方法，也可用于你自己的研究。