InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now. Learn more →
Top 23 Python Jax Projects
-
Project mention: None of the top 10 projects in GitHub is actually a software project 🤯 | dev.to | 2025-05-10
We see an addition to the AI community with AutoGPT. Along with Tensorflow they represent the AI community in the software category, which is getting relevant (2 out of 8). We can expect in the future to have new AI projects in the top 25 such as Transformers or Ollama (currently top 34 and 36, respectively).
-
InfluxDB
InfluxDB – Built for High-Performance Time Series Workloads. InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.
-
The unchallenged leader in AI development is still Python. and Keras, and robust community support.
-
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
I'm not a fan of Python, but it is the languange of choice for most Machine Learning code. I'm currently working on a NodeJS based project where I need to call the HuggingFace Diffusers python module. So I came up with an IPC wrapper around the the Python script.
-
d2l-en
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
-
Project mention: A ranked list of machine learning Python libraries. Updated weekly | news.ycombinator.com | 2025-01-31
-
-
SaaSHub
SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives
-
wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
Project mention: 💡 TinyLlama Meets LoRA: A Lightweight Approach to Emotion Classification | dev.to | 2025-05-05Tweak Hyperparameters: I’d love to use Weights & Biases to fine-tune settings.
-
einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
-
-
-
-
Good read! I am developing PINNs at work and this certainly helped me recall important concepts. This post used deepxde library [2] to compose the PINN. Can anyone comment on how NVIDIA's modulus [2] compares to this? Modulus appears to be much more verbose and poorly documented.
[1]: https://github.com/lululxvi/deepxde
-
-
-
foolbox
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
-
pennylane
PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.
Penny Lane: https://pennylane.ai/
-
EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
-
-
numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Project mention: Numpyro: Probabilistic programming with NumPy powered by Jax | news.ycombinator.com | 2024-11-16 -
TransformerEngine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.
Project mention: TransformerEngine: A library for accelerating Transformer models on Nvidia GPUs | news.ycombinator.com | 2024-09-25 -
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Project mention: An Introduction to Neural Ordinary Differential Equations [pdf] | news.ycombinator.com | 2025-01-11 -
Project mention: AXLearn: Apple's Deep Learning library built on top of Jax | news.ycombinator.com | 2025-02-18
-
SaaSHub
SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives
NOTE:
The open source projects on this list are ordered by number of github stars.
The number of mentions indicates repo mentiontions in the last 12 Months or
since we started tracking (Dec 2020).
Python Jax discussion
Python Jax related posts
-
Show HN: Localscope–Limit scope of Python functions for reproducible execution
-
AXLearn: Apple's Deep Learning library built on top of Jax
-
How to Install & Run VideoLLaMA3-7B Locally
-
Bullish on AI infrastructure, bearish on AI developer frameworks
-
Show HN: Design/build of some elegant parametric speaker cabinets with OpenSCAD
-
An Introduction to Neural Ordinary Differential Equations [pdf]
-
KlongPy: High-Performance Array Programming in Python
-
A note from our sponsor - InfluxDB
www.influxdata.com | 16 May 2025
Index
What are some of the best open-source Jax projects in Python? This list will help you:
| # | Project | Stars |
|---|---|---|
| 1 | transformers | 144,375 |
| 2 | Keras | 62,989 |
| 3 | jax | 32,213 |
| 4 | diffusers | 29,010 |
| 5 | d2l-en | 25,776 |
| 6 | best-of-ml-python | 20,054 |
| 7 | ivy | 14,202 |
| 8 | wandb | 9,847 |
| 9 | einops | 8,918 |
| 10 | trax | 8,204 |
| 11 | datasets | 4,409 |
| 12 | scenic | 3,538 |
| 13 | deepxde | 3,204 |
| 14 | dm-haiku | 3,024 |
| 15 | thinc | 2,851 |
| 16 | foolbox | 2,846 |
| 17 | pennylane | 2,628 |
| 18 | EasyLM | 2,476 |
| 19 | mctx | 2,480 |
| 20 | numpyro | 2,444 |
| 21 | TransformerEngine | 2,411 |
| 22 | equinox | 2,352 |
| 23 | axlearn | 2,051 |
Sponsored
InfluxDB – Built for High-Performance Time Series Workloads
InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.
www.influxdata.com