Projects Using JAX for Accelerated Machine Learning, Differentiable Programming, and High-Performance Numerical Computing
Explore projects tagged with jax to find open-source and enterprise projects that leverage JAX for accelerated machine learning, differentiable programming, autograd pipelines, and XLA-optimized numerical computing on GPU/TPU. This curated list displays items from the projects nav filtered by the tags pillar for the jax item, highlighting real-world use cases such as scalable training loops, probabilistic modeling, custom gradients, and production-ready inference workflows. Use the filtering UI to narrow results by domain, hardware, license, or maturity, then compare repositories, performance benchmarks, and deployment patterns to identify solutions to adopt or contribute to. Browse these jax-tagged projects now to evaluate performance, view source code, and take action—fork, contribute, or integrate proven JAX implementations into your stack.
