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 16 Jupyter Notebook Jax Projects
-
-
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.
-
-
-
uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
-
-
score_sde
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
-
https://github.com/google/jax-cfd#other-awesome-projects
PhiFlow:
-
SaaSHub
SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives
-
tensor-sensor
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
-
get-started-with-JAX
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
-
jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
-
-
-
-
jaxrenderer
Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrenderer/wiki; PR: https://github.com/google/brax/pull/367
-
-
-
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).
Jupyter Notebook Jax discussion
Jupyter Notebook Jax related posts
-
Olelo Foil: NACA Airfoil SIM
-
Bayesian Analysis with Python
-
How to stay up-to-date with the latest AI company announcements and events?
-
Physic engine for 3D simulation: which one to use?
-
Tips to become self taught machine learning engineer
-
Jax-CFD: Computational Fluid Dynamics in Jax
-
Deep Learning Poised to ‘Blow Up’ Famed Fluid Equations
-
A note from our sponsor - InfluxDB
www.influxdata.com | 21 May 2025
Index
What are some of the best open-source Jax projects in Jupyter Notebook? This list will help you:
| # | Project | Stars |
|---|---|---|
| 1 | pyprobml | 6,765 |
| 2 | flax | 6,549 |
| 3 | whisper-jax | 4,577 |
| 4 | uvadlc_notebooks | 2,841 |
| 5 | neural-tangents | 2,322 |
| 6 | score_sde | 1,578 |
| 7 | jax-cfd | 835 |
| 8 | tensor-sensor | 805 |
| 9 | get-started-with-JAX | 693 |
| 10 | jaxrl | 673 |
| 11 | TF_JAX_tutorials | 275 |
| 12 | indaba-pracs-2022 | 176 |
| 13 | hyper-nn | 100 |
| 14 | jaxrenderer | 72 |
| 15 | flock_env | 13 |
| 16 | forwardalgorithms | 10 |
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