Awesome

GitHub projects from awesome lists

Search awesome repositories

Search names, descriptions, topics, tags, and stacks, then tune results by ecosystem, freshness, health, and cross-list signal.

Continue with GitHub Browse awesome lists Request a list

Repos indexed: 17,373
Awesome lists tracked: 125
Current results: 16

Find repositories

Start broad, then narrow by ecosystem, freshness, health, and growth.

Clear 1 refinement

Search repositories

Search mode

Keyword Semantic

Tune results

The controls most people need first.

Awesome list

Language

Freshness

Sort

Direction

More filters Topics, generated tags, stack, files, age, archive status, and growth.

Ecosystem

GitHub topic

Generated tag

Framework or stack

Package manager

Files

Has file

Choose a suggestion or use commas to require multiple files.

Health

Minimum stars

Repository age

Uses known first-commit dates.

Archive status

AI development signals

Momentum

Unmaintained for

Commit velocity

Star growth

Reset filters

16 repos shown

Topic: jax

Browse

Highlighted

Open highlighted repo slot

Put your repository first

Promote a GitHub repo at the top of Awesome repository list views for 7 days.

keras-team/keras

Deep Learning for humans

AI dev

Stack

Python pytest PEP 517 pip

GitHub topics

#data-science #deep-learning #jax #machine-learning #neural-networks #python

Updated: 2026-07-16
Lists: 4 list mentions
First commit: 2015-03-28
History: 60 history points
License: Apache-2.0
Issues: 212 open

64,169

stars

Forks: 19,739
Commits: 12,542 commits
Star growth, last 7 days: -18 0.0%
Commit velocity, last 7 days: +23 +0.2%

Website GitHub

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Stack

Python pytest CMake PEP 517 pip

GitHub topics

#jax

Updated: 2026-07-17
Lists: 3 list mentions
First commit: 2018-11-18
History: 52 history points
License: Apache-2.0
Issues: 2,478 open

36,008

stars

Forks: 3,682
Commits: 37,080 commits
Star growth, last 7 days: +10 0.0%
Commit velocity, last 7 days: +163 +0.4%

Website GitHub

lukasmasuch/best-of-ml-python

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

GitHub topics

#automl #chatgpt #data-analysis #data-science #data-visualization #data-visualizations

Updated: 2026-07-02
Lists: 2 list mentions
First commit: 2020-11-29
History: 4 history points
License: CC-BY-SA-4.0
Issues: 50 open

23,659

stars

Forks: 3,138
Commits: 628 commits
Star growth, last 7 days: No 7-day history
Commit velocity, last 7 days: No 7-day history

Website GitHub

arogozhnikov/einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Stack

Python uv

GitHub topics

#cupy #deep-learning #einops #jax #mlx #numpy

Updated: 2026-07-05
Lists: 2 list mentions
First commit: 2018-09-22
History: 5 history points
License: MIT
Issues: 39 open

9,549

stars

Forks: 400
Commits: 747 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

HIPS/autograd

Efficiently computes derivatives of NumPy code.

Stack

Python PEP 517

GitHub topics

#autograd #automatic-differentiation #backpropagation #deep-learning #derivative #differentiation

Updated: 2026-07-02
Lists: 1 list mention
First commit: 2014-11-24
History: 6 history points
License: MIT
Issues: 183 open

7,508

stars

Forks: 937
Commits: 1,561 commits
Star growth, last 7 days: No 7-day history
Commit velocity, last 7 days: No 7-day history

GitHub

google/flax

Flax is a neural network library for JAX that is designed for flexibility.

Stack

Jupyter Notebook pytest Cargo CMake PEP 517

GitHub topics

#jax

Updated: 2026-07-07
Lists: 2 list mentions
First commit: 2020-01-07
History: 7 history points
License: Apache-2.0
Issues: 493 open

7,267

stars

Forks: 824
Commits: 5,891 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

iree-org/iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Stack

C++ pytest CMake PEP 517 pip

GitHub topics

#compiler #cuda #jax #machine-learning #mlir #onnx

Updated: 2026-07-14
Lists: 1 list mention
First commit: 2019-09-18
History: 5 history points
License: Apache-2.0
Issues: 1,654 open

3,840

stars

Forks: 952
Commits: 20,310 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

PennyLaneAI/pennylane

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

Stack

Python pytest pip uv

GitHub topics

#autograd #automatic-differentiation #cirq #deep-learning #differentiable-computing #jax

Updated: 2026-07-08
Lists: 2 list mentions
First commit: 2018-04-17
History: 9 history points
License: Apache-2.0
Issues: 394 open

3,291

stars

Forks: 823
Commits: 7,048 commits
Star growth, last 7 days: No 7-day history
Commit velocity, last 7 days: No 7-day history

Website GitHub

google-deepmind/dm-haiku

JAX-based neural network library

Stack

Python pip

GitHub topics

#deep-learning #deep-neural-networks #jax #machine-learning #neural-networks

Updated: 2026-07-09
Lists: 1 list mention
First commit: 2020-01-22
History: 5 history points
License: Apache-2.0
Issues: Disabled

3,256

stars

Forks: 300
Commits: 1,064 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Stack

Python pytest uv

GitHub topics

#deep-learning #equinox #jax #neural-networks

Updated: 2026-06-29
Lists: 1 list mention
First commit: 2021-07-28
History: 5 history points
License: Apache-2.0
Issues: 244 open

2,923

stars

Forks: 204
Commits: 1,131 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

pyro-ppl/numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Stack

Python Jupyter pytest uv

GitHub topics

#bayesian-inference #hmc #inference-algorithms #jax #mcmc #numpy

Updated: 2026-07-09
Lists: 1 list mention
First commit: 2019-02-15
History: 5 history points
License: Apache-2.0
Issues: 66 open

2,718

stars

Forks: 293
Commits: 1,432 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

patrick-kidger/diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Stack

Python pytest uv

GitHub topics

#deep-learning #differential-equations #dynamical-systems #equinox #jax #machine-learning

Updated: 2026-06-21
Lists: 1 list mention
First commit: 2021-07-25
History: 5 history points
License: Apache-2.0
Issues: 236 open

2,066

stars

Forks: 186
Commits: 729 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

kubeflow/katib

Automated Machine Learning on Kubernetes

Stack

Python Go modules npm PEP 517

GitHub topics

#ai #automl #huggingface #hyperparameter-tuning #jax #kubeflow

Updated: 2026-07-14
Lists: 1 list mention
First commit: 2018-04-03
History: 5 history points
License: Apache-2.0
Issues: 105 open

1,690

stars

Forks: 533
Commits: 1,376 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

google-deepmind/jraph

A Graph Neural Network Library in Jax

Archived

Stack

Python pip

GitHub topics

#deep-learning #graph-neural-networks #jax #machine-learning

Updated: 2024-03-18
Lists: 1 list mention
First commit: 2020-10-20
History: 6 history points
License: Apache-2.0
Issues: 19 open

1,471

stars

Forks: 106
Commits: 103 commits
Star growth, last 7 days: No 7-day history
Commit velocity, last 7 days: No 7-day history

Website GitHub

Toni-SM/skrl

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

Stack

Python pytest PEP 517 pip

GitHub topics

#brax #deep-learning #flax #gym #gymnasium #isaaclab

Updated: 2026-05-11
Lists: 2 list mentions
First commit: 2021-07-08
History: 7 history points
License: MIT
Issues: 28 open

1,082

stars

Forks: 150
Commits: 1,570 commits
Star growth, last 7 days: 0 0.0%
Commit velocity, last 7 days: 0 0.0%

Website GitHub

keras-team/keras-hub

Pretrained model hub for Keras 3.

AI dev

Stack

Python PEP 517 pip

GitHub topics

#cv #deep-learning #jax #keras #llm #machine-learning

Updated: 2026-07-02
Lists: 1 list mention
First commit: 2020-09-12
History: 6 history points
License: Apache-2.0
Issues: 245 open

986

stars

Forks: 352
Commits: 1,548 commits
Star growth, last 7 days: No 7-day history
Commit velocity, last 7 days: No 7-day history

Website GitHub

Search awesome repositories

Find repositories

Put your repository first

How it works

Pricing

How it works

Pricing