Awesome Open Source

Search

Programming Languages

About

The Top 10 Jax Open Source Projects

Open source projects categorized as Jax

Categories > Jax

Edit Category

huggingface/transformers ⭐ 119,240

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

dependent packages 0 total releases 0 most recent commit over 2 years ago

keras-team/keras ⭐ 60,198

Deep Learning for humans

dependent packages 0 total releases 0 most recent commit over 2 years ago

d2l-ai/d2l-en ⭐ 20,613

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

dependent packages 0 total releases 0 most recent commit over 2 years ago

ml-tooling/best-of-ml-python ⭐ 14,990

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

dependent packages 0 total releases 0 most recent commit over 2 years ago

unifyai/ivy ⭐ 13,904

The Unified AI Framework

dependent packages 0 total releases 0 most recent commit over 2 years ago

google/trax ⭐ 7,818

Trax — Deep Learning with Clear Code and Speed

dependent packages 0 total releases 0 most recent commit over 2 years ago

wandb/wandb ⭐ 7,760

🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.

dependent packages 0 total releases 0 most recent commit over 2 years ago

arogozhnikov/einops ⭐ 7,548

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

dependent packages 0 total releases 0 most recent commit over 2 years ago

HIPS/autograd ⭐ 7,506

Efficiently computes derivatives of NumPy code.

dependent packages 0 total releases 0 most recent commit 12 days ago

probml/pyprobml ⭐ 6,066

Python code for "Probabilistic Machine learning" book by Kevin Murphy

dependent packages 0 total releases 0 most recent commit over 2 years ago

Get A Weekly Email With Trending Jax Projects

No Spam. Unsubscribe easily at any time.

Privacy | About | Terms | Follow Us On Twitter

Downloads, Dependent Repos, Dependent Packages, Total Releases, Latest Releases data powered by Libraries.io.

Copyright 2018-2026 Awesome Open Source. All rights reserved.