Google 的端到端机器学习平台,生态完整。
⚙️ 深度学习基础库
PyTorch、Transformers、JAX 等底层框架与工具
数万预训练模型的统一接口,覆盖文本、图像、音频与多模态。
动态计算图的深度学习框架,研究与工业界的主流选择。
Python 经典机器学习库,简单高效。
多后端(JAX / TF / PyTorch)的深度学习 API。
最简单快速的中型 GPT 训练与微调仓库。
百元级预算从零训练 ChatGPT 式模型的极简全流程。
可组合变换(JIT、vmap、grad)的高性能数值计算库。
💫 Industrial-strength Natural Language Processing (NLP) in Python
极简的深度学习框架,介于 PyTorch 与 micrograd 之间。
Visualizer for neural network, deep learning and machine learning models
可扩展的梯度提升决策树库。
The fastai deep learning library
A game theoretic approach to explain the output of any machine learning model.
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
快速且省显存的精确注意力实现。
百度飞桨,产业级深度学习平台。
Graph Neural Network Library for PyTorch
A reactive notebook for Python — run reproducible experiments, query with SQL, execute as a script, deploy as an app, and version with git. Stored as pure Python. All in a modern, AI-native editor.
跨平台高性能机器学习推理与训练加速器。
机器学习模型的开放互操作格式。
用 Python 写高效 GPU 内核的语言与编译器。
OpenAI 模型使用的快速 BPE 分词器。
高效的梯度提升框架。