PyTorch

Dynamic computation graphs, Pythonic API, dominant in research and production. The current standard for most frontier AI work.

View on GitHub Website Active

Guide pending

We haven't written this guide yet.

Guides are drafted from the project's own README and metadata. Until one exists, check the product's repository and documentation for authoritative information.

Nearby on the map

Related projects

TensorFlow

tensorflow/tensorflow

End-to-end platform with excellent production deployment, TPU support, and large-scale serving tools.

201k 79kC++pushed 2 hours ago

dm-haiku

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.

3.3k 302Pythonpushed 3 days ago

Diffrax

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.

2.1k 199Pythonpushed 2 months ago

Equinox

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.

3k 221Pythonpushed 5 days ago

JAX

jax-ml/jax

High-performance numerical computing with composable transformations (JIT, vmap, grad). Rising favorite for research and scientific ML.

37k 4kPythonpushed 2 hours ago

vit-pytorch

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.

26k 3.5kPythonpushed 5 days ago