Google's end-to-end machine learning platform with a complete ecosystem.
⚙️ Deep Learning Libraries
PyTorch, Transformers, JAX and other low-level frameworks and tools
A unified interface to tens of thousands of pre-trained models covering text, images, audio and multimodal tasks.
A deep-learning framework with dynamic computation graphs, the mainstream choice in both research and industry.
The classic Python machine-learning library: simple and efficient.
多后端(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 Modular Platform (includes MAX & Mojo)
可扩展的梯度提升决策树库。
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 分词器。