tensorflow/tensorflow
End-to-end platform with excellent production deployment, TPU support, and large-scale serving tools.
ollama/ollama
Dead-simple local LLM runner with a one-line install, model registry, and OpenAI-compatible API.
langchain-ai/langchain
Foundational library for agents, chains, and memory.
ggml-org/llama.cpp
Pure C/C++ inference engine with GGUF format support. The gold standard for CPU/GPU/Apple Silicon on-device running. Includes llama-server for OpenAI-compatible API. Now at 100K+ stars.
pytorch/pytorch
Dynamic computation graphs, Pythonic API, dominant in research and production. The current standard for most frontier AI work.
vllm-project/vllm
State-of-the-art serving engine with PagedAttention and continuous batching. Currently the fastest production-grade LLM server.
jax-ml/jax
High-performance numerical computing with composable transformations (JIT, vmap, grad). Rising favorite for research and scientific ML.
lucidrains/vit-pytorch
Comprehensive Vision Transformer (ViT) implementations in PyTorch. Reference implementations of all major vision transformer variants including ViT, DeiT, Swin, and more. MIT licensed.
google/flax
Neural network library for JAX, designed for flexibility. Apache-2.0 licensed.
google-deepmind/dm-haiku
JAX-based neural network library from Google DeepMind. Elegant functional API with state management, widely used in DeepMind's research. Apache 2.0 licensed.
patrick-kidger/equinox
Elegant easy-to-use neural networks and scientific computing in JAX. Callable PyTrees with filtered transformations, seamless interoperability with the JAX ecosystem. Apache 2.0 licensed.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable ODE/SDE/CDE solvers for scientific machine learning and neural differential equations. Apache 2.0 licensed.
pyro-ppl/numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation. Bayesian modeling and inference at scale.
keras-team/keras
High-level, beginner-friendly API that now runs on multiple backends (TensorFlow, JAX, PyTorch). Perfect for rapid experimentation.
tinygrad/tinygrad
Minimalist deep learning framework with tiny code footprint. The "you like PyTorch? you like micrograd? you love tinygrad!" philosophy - simple yet powerful.
PaddlePaddle/Paddle
Industrial deep learning platform from Baidu serving 23+ million developers and 760,000+ companies. China's first independent R&D framework with advanced distributed training and deployment capabilities.
pyg-team/pytorch_geometric
Library for deep learning on irregular input data such as graphs, point clouds, and manifolds. Part of the PyTorch ecosystem.
huggingface/pytorch-image-models
The largest collection of PyTorch image encoders and backbones. 900+ pretrained models including ResNet, EfficientNet, Vision Transformer, ConvNeXt, and more with training and inference scripts. Apache 2.0 licensed.
triton-lang/triton
Language and compiler for writing highly efficient custom deep-learning primitives. Powers kernel optimizations in PyTorch, JAX, and other frameworks. MIT licensed.
ggml-org/ggml
Tensor library for machine learning. The foundational C/C++ library powering llama.cpp and many on-device inference engines. MIT licensed.
ml-explore/mlx
Array framework for machine learning on Apple silicon. Efficient unified memory design with NumPy-like API, automatic differentiation, and multi-device support. MIT licensed.
ariannamethod/notorch
Neural network framework in pure C with reverse-mode automatic differentiation, model training, and GGUF inference, without a Python runtime.
uxlfoundation/oneDNN
oneAPI Deep Neural Network Library. Cross-platform performance library of basic building blocks for deep learning, optimized for Intel CPUs, GPUs, and Arm architectures. Apache 2.0 licensed.
onnx/onnx
Open standard for machine learning interoperability. Open Neural Network Exchange provides an open ecosystem that empowers AI developers to choose the right tools as their project evolves. Apache 2.0 licensed.