Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
☆10Jul 27, 2026Updated this week
Alternatives and similar repositories for jax
Users that are interested in jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Scipy main repository☆13Updated this week
- ☆17Sep 24, 2023Updated 2 years ago
- Numpy main repository☆14Oct 22, 2025Updated 9 months ago
- Probabilistic reasoning and statistical analysis in TensorFlow☆12Aug 27, 2021Updated 4 years ago
- Get unique objects from an array of objects based on single/multiple keys.☆11Mar 26, 2026Updated 4 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- PhysAI is an open-source AI project that aims to link quantum mechanics and general relativity by generating, testing, and improving phys…☆16Nov 10, 2025Updated 8 months ago
- Jupyter Notebooks with Snowpark☆15Nov 3, 2021Updated 4 years ago
- Optimizers in tensorflow from scratch☆18Jun 6, 2017Updated 9 years ago
- In this repo, you can find all python problems for the Coding Ninja Fundamental course of 2021-22.☆18Jan 1, 2022Updated 4 years ago
- PPO and PyMARL baseline for Pogema environment☆24Sep 18, 2024Updated last year
- Jupyter Notebooks and Virtual Environments☆21Jun 5, 2023Updated 3 years ago
- Book / Blog of different topics around MLOps engineering.☆11Jul 6, 2026Updated 3 weeks ago
- A numpyro implementation of BayeSN, the optical-NIR SED model for type Ia supernovae☆18Jun 16, 2026Updated last month
- Dataiku DSS plugin to automate time series forecasting with Deep Learning and statistical models 📈☆17Apr 14, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A custom transfer agent for git-lfs that uses rclone to transfer files.☆13Jan 14, 2025Updated last year
- implementation of replicated softmax model☆13Sep 25, 2014Updated 11 years ago
- An unofficial jax/haiku implementation of Crystal Graph Convolutional Neural Networks (CGCNN)☆10Dec 17, 2022Updated 3 years ago
- ZTF Uniform Depth Survey Image Processing Pipeline☆13Apr 13, 2026Updated 3 months ago
- A Python library for pruning and visualizing Keras Neural Networks' structure and weights☆10Dec 24, 2019Updated 6 years ago
- Age classification from text using PAN16, blogs, Fisher Callhome, and Cancer Forum☆18Jul 1, 2022Updated 4 years ago
- JAX implementation and extension of the healpy library☆15Jul 7, 2026Updated 3 weeks ago
- "A Discrete Variational Recurrent Topic Model without the Reparametrization Trick" (NeurIPS 2020)☆11Apr 26, 2021Updated 5 years ago
- An Implementation of Mutli-layer Perceptron in C++☆10Jul 14, 2021Updated 5 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Solutions of all python problems of coursera MOOC python basic by University of Michigan under course Python 3 Prgramming☆17Jul 19, 2020Updated 6 years ago
- Testing of Neural Topic Modeling for Japanese articles☆13Jul 24, 2019Updated 7 years ago
- 「M1 MacにPythonの環境構築してみた」動画の資料です☆16Mar 24, 2021Updated 5 years ago
- Physics-Informed Neural Network☆95May 10, 2024Updated 2 years ago
- Text Summarization using Transformer on GPU Docker Deployment☆15Jul 26, 2022Updated 4 years ago
- Simulate transients within the sky☆16May 25, 2026Updated 2 months ago
- SoftEther VPN (ALLFREE Edition Repository): Open Cross-platform Multi-protocol VPN Software. This branch is remove function restrictions …☆10Dec 1, 2022Updated 3 years ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Oct 28, 2025Updated 9 months ago
- Mini-batch multiplicative updates for beta-NMF☆15Jun 27, 2017Updated 9 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Local display of a Jupyter notebook running on a distant server☆14Jun 5, 2023Updated 3 years ago
- This Project provides a sample implementation that will show how to leverage Amazon Athena from .NET Core Application using AWS SDK for .…☆24Oct 5, 2023Updated 2 years ago
- Python-based cryptocurrency trading bot that functions on the Coinbase Pro exchange. This bot currently uses RSI values and 24HR Volume i…☆10May 16, 2022Updated 4 years ago
- Python time-and-the-sky programs built mostly on astropy; packaged for installation.☆21Jul 11, 2020Updated 6 years ago
- RingCentral Softphone SDK for JavaScript