🧱 深度学习框架
DL Frameworks
构建和训练神经网络的核心工具。PyTorch在学术界和工业界占据主导地位,TensorFlow在端侧部署仍有优势,PaddlePaddle和MindSpore是国产框架代表。框架选择直接影响开发效率、模型性能和部署路径。
平台/工具(6项)
主流框架
🌍 海外
JAX
Google开源的函数式数值计算框架,支持自动微分和JIT编译到XLA后端。以纯函数、向量化变换(vmap/pmap)和函数式风格著称。DeepMind大量使用JAX进行前沿研究,Google Gem...
MXNet
Apache基金会开源深度学习框架,以轻量级和高效分布式训练著称。曾被AWS选为官方深度学习框架,支撑Amazon SageMaker服务。支持多语言API(Python/Java/Scala/C++...
PyTorch
Meta AI开源的深度学习框架,以动态计算图和Python优先的设计理念成为学术界和工业界首选。TorchDynamo编译器和FSDP分布式策略支持大规模训练。在HuggingFace、OpenAI...
TensorFlow
Google开源的深度学习框架,以静态计算图和强大生产部署能力为核心特色。TensorFlow Serving支持高性能推理服务,TFLite覆盖移动端/AIoT场景。虽然研究领域份额被PyTorch...