The Top 23 Python Jax Open Source Projects

Open source projects categorized as Python Jax

Categories > Jax

Categories > Programming Languages > Python

Transformers ⭐ 124,049

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

dependent packages 2,484total releases 125most recent commit 15 days ago

Keras ⭐ 60,854

Deep Learning for humans

dependent packages 697total releases 87most recent commit 14 days ago

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.

total releases 2most recent commit 3 months ago

Best Of Ml Python ⭐ 14,990

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

most recent commit 3 months ago

Ivy ⭐ 13,904

The Unified AI Framework

dependent packages 7total releases 103most recent commit 3 months ago

Trax ⭐ 7,818

Trax — Deep Learning with Clear Code and Speed

dependent packages 1total releases 24most recent commit 5 months ago

Einops ⭐ 7,548

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

dependent packages 767total releases 14most recent commit 3 months ago

Flax ⭐ 5,144

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

dependent packages 188total releases 40most recent commit 3 months ago

Datasets ⭐ 4,094

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

dependent packages 163total releases 36most recent commit 3 months ago

Interviews.ai ⭐ 3,146

It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.

most recent commit 2 years ago

Alpa ⭐ 2,878

Training and serving large-scale neural networks with auto parallelization.

total releases 13most recent commit 5 months ago

Thinc ⭐ 2,774

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

dependent packages 67total releases 229most recent commit 3 months ago

Scenic ⭐ 2,733

Scenic: A Jax Library for Computer Vision Research and Beyond

most recent commit 3 months ago

Foolbox ⭐ 2,600

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

dependent packages 5total releases 70most recent commit 3 months ago

Deepxde ⭐ 2,135

A library for scientific machine learning and physics-informed learning

dependent packages 3total releases 61most recent commit 3 months ago

Easylm ⭐ 2,087

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

most recent commit 4 months ago

Pennylane ⭐ 2,022

PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.

dependent packages 54total releases 46most recent commit 3 months ago

Numpyro ⭐ 1,966

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

dependent packages 72total releases 26most recent commit 3 months ago

Equinox ⭐ 1,614

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

dependent packages 44total releases 46most recent commit 4 months ago

Machine_learning_refined ⭐ 1,574

Notes, examples, and Python demos for the 2nd edition of the textbook "Machine Learning Refined" (published by Cambridge University Press).

most recent commit 10 months ago

Tensorly ⭐ 1,466

TensorLy: Tensor Learning in Python.

dependent packages 39total releases 20most recent commit 4 months ago

Training Operator ⭐ 1,447

Distributed ML Training and Fine-Tuning on Kubernetes

dependent packages 28total releases 128most recent commit 3 days ago

Big_transfer ⭐ 1,421

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

most recent commit a year ago

The 10 Most Depended On Python Jax Open Source Projects

Transformers ⭐ 124,049

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

dependent packages 2,484total releases 125latest release November 15, 2023most recent commit 15 days ago

Einops ⭐ 7,548

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

dependent packages 767total releases 14latest release October 01, 2023most recent commit 3 months ago

Keras ⭐ 60,854

Deep Learning for humans

dependent packages 697total releases 87latest release December 06, 2023most recent commit 14 days ago

Flax ⭐ 5,144

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

dependent packages 188total releases 40latest release October 28, 2023most recent commit 3 months ago

Datasets ⭐ 4,094

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

dependent packages 163total releases 36latest release September 08, 2023most recent commit 3 months ago

Numpyro ⭐ 1,966

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

dependent packages 72total releases 26latest release September 24, 2023most recent commit 3 months ago

Jaxtyping ⭐ 759

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

dependent packages 71total releases 28latest release November 27, 2023most recent commit 3 months ago

Thinc ⭐ 2,774

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

dependent packages 67total releases 229latest release September 22, 2023most recent commit 3 months ago

Pennylane ⭐ 2,022

PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.

dependent packages 54total releases 46latest release November 08, 2023most recent commit 3 months ago

Equinox ⭐ 1,614

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

dependent packages 44total releases 46latest release November 13, 2023most recent commit 4 months ago

The 10 Latest Releases In Python Jax Open Source Projects

Jaxlie ⭐ 192

Rigid transforms + Lie groups, in JAX

dependent packages 1total releases 26latest release December 11, 2023most recent commit 4 months ago

Orbax ⭐ 187

Orbax provides common utility libraries for JAX users.

dependent packages 9total releases 28latest release December 07, 2023most recent commit 3 months ago

Opennsfw2 ⭐ 272

Keras implementation of the Yahoo Open-NSFW model

total releases 23latest release December 07, 2023most recent commit 4 months ago

Keras ⭐ 60,854

Deep Learning for humans

dependent packages 697total releases 87latest release December 06, 2023most recent commit 14 days ago

Mosec ⭐ 661

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

dependent packages 1total releases 40latest release December 05, 2023most recent commit 3 months ago

Dm_pix ⭐ 352

PIX is an image processing library in JAX, for JAX.

total releases 11latest release December 04, 2023most recent commit 8 days ago

Ninjax ⭐ 55

General Modules for JAX

total releases 27latest release November 30, 2023most recent commit 5 months ago

Evox ⭐ 98

Distributed GPU-Accelerated Framework for Evolutionary Computation. Comprehensive Library of Evolutionary Algorithms & Benchmark Problems.

total releases 10latest release November 30, 2023most recent commit 3 months ago

Gpjax ⭐ 352

Gaussian processes in JAX.

dependent packages 5total releases 55latest release November 27, 2023most recent commit 3 months ago

Jaxtyping ⭐ 759

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

dependent packages 71total releases 28latest release November 27, 2023most recent commit 3 months ago