Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About

The Top 23 Jax Open Source Projects

Open source projects categorized as Jax
Categories > Jax
Transformers ⭐ 124,049
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
dependent packages 2,484total releases 125most recent commit 14 days agopypi transformers} Downloads
Keras ⭐ 60,854
Deep Learning for humans
dependent packages 697total releases 87most recent commit 13 days agopypi keras} Downloads
Jax ⭐ 27,067
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
dependent packages 743total releases 155most recent commit a month agopypi jax} Downloads
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 agopypi ivy} Downloads
Wandb ⭐ 8,204
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
dependent packages 622total releases 266most recent commit 17 hours agopypi wandb} Downloads
Trax ⭐ 7,818
Trax — Deep Learning with Clear Code and Speed
dependent packages 1total releases 24most recent commit 5 months agopypi trax} Downloads
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 agopypi einops} Downloads
Pyprobml ⭐ 6,066
Python code for "Probabilistic Machine learning" book by Kevin Murphy
most recent commit 4 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 agopypi flax} Downloads
Datasets ⭐ 4,094
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
dependent packages 163total releases 36most recent commit 3 months agopypi tensorflow-datasets} Downloads
Whisper Jax ⭐ 3,824
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
most 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
Dm Haiku ⭐ 2,807
JAX-based neural network library
dependent packages 87total releases 14most recent commit 15 days agopypi dm-haiku} Downloads
Thinc ⭐ 2,774
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
dependent packages 67total releases 229most recent commit 3 months agopypi thinc} Downloads
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 agopypi foolbox} Downloads
Faust ⭐ 2,351
Functional programming language for signal processing and sound synthesis
dependent packages 3total releases 7most recent commit 3 months agonpm @grame/libfaust} Downloads
Iree ⭐ 2,303
A retargetable MLIR-based machine learning compiler and runtime toolkit.
total releases 17most recent commit a month ago
Mctx ⭐ 2,201
Monte Carlo tree search in JAX
dependent packages 1total releases 4most recent commit 3 months agopypi mctx} Downloads
Neural Tangents ⭐ 2,149
Fast and Easy Infinite Neural Networks in Python
dependent packages 1total releases 30most recent commit 4 months agopypi neural-tangents} Downloads
The 10 Most Depended On 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 14 days agopypi transformers} Downloads
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 agopypi einops} Downloads
Jax ⭐ 27,067
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
dependent packages 743total releases 155latest release December 04, 2023most recent commit a month agopypi jax} Downloads
Keras ⭐ 60,854
Deep Learning for humans
dependent packages 697total releases 87latest release December 06, 2023most recent commit 13 days agopypi keras} Downloads
Wandb ⭐ 8,204
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
dependent packages 622total releases 266latest release December 05, 2023most recent commit 17 hours agopypi wandb} Downloads
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 agopypi flax} Downloads
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 agopypi tensorflow-datasets} Downloads
Dm Haiku ⭐ 2,807
JAX-based neural network library
dependent packages 87total releases 14latest release November 10, 2023most recent commit 15 days agopypi dm-haiku} Downloads
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 agopypi numpyro} Downloads
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 agopypi jaxtyping} Downloads
The 10 Latest Releases In Jax Open Source Projects
Mpi4jax ⭐ 348
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
dependent packages 2total releases 61latest release December 11, 2023most recent commit 2 months agopypi mpi4jax} Downloads
Jaxlie ⭐ 192
Rigid transforms + Lie groups, in JAX
dependent packages 1total releases 26latest release December 11, 2023most recent commit 4 months agopypi jaxlie} Downloads
Orbax ⭐ 187
Orbax provides common utility libraries for JAX users.
dependent packages 9total releases 28latest release December 07, 2023most recent commit 3 months agopypi orbax-checkpoint} Downloads
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 13 days agopypi keras} Downloads
Wandb ⭐ 8,204
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
dependent packages 622total releases 266latest release December 05, 2023most recent commit 17 hours agopypi wandb} Downloads
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 agopypi mosec} Downloads
Jax ⭐ 27,067
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
dependent packages 743total releases 155latest release December 04, 2023most recent commit a month agopypi jax} Downloads
Dm_pix ⭐ 352
PIX is an image processing library in JAX, for JAX.
total releases 11latest release December 04, 2023most recent commit 7 days ago
Ninjax ⭐ 55
General Modules for JAX
total releases 27latest release November 30, 2023most recent commit 5 months ago
Categories
Privacy | About | Terms | Follow Us On Twitter

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

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