NumPyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation. Bayesian modeling and inference at scale.

View on GitHub Unknown

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

PyTorch

pytorch/pytorch

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

105k 32kPythonpushed 2 hours 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

Flax

google/flax

Neural network library for JAX, designed for flexibility. Apache-2.0 licensed.

7.4k 848Jupyter Notebookpushed 4 days 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