A stateful pytree library for training neural networks.
☆22Aug 31, 2025Updated last year
Alternatives and similar repositories for pax
Users that are interested in pax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- AlphaZero in JAX☆82Apr 3, 2024Updated 2 years ago
- Stencil computations in JAX☆73Nov 15, 2025Updated 9 months ago
- Using JAX to generate piano music as MIDI☆39Nov 28, 2023Updated 2 years ago
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆72Jul 24, 2022Updated 4 years ago
- Butterfly Reader for Hacker News, built with Flutter☆16Oct 27, 2019Updated 6 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Evaluate your models with A/B test experiments☆14Jan 5, 2023Updated 3 years ago
- JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library☆15Oct 12, 2025Updated 10 months ago
- A Pytree Module system for Deep Learning in JAX☆210Feb 26, 2023Updated 3 years ago
- Compiler for the Albatross Language☆12Sep 29, 2021Updated 4 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 3 years ago
- nashorn websockets demo☆16Oct 2, 2013Updated 12 years ago
- ☆11May 12, 2021Updated 5 years ago
- If it quacks like a tensor...☆59Nov 13, 2024Updated last year
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 3 years ago
- A functional training loops library for JAX☆87Feb 13, 2024Updated 2 years ago
- Toggl Plan browser extension☆16Sep 27, 2023Updated 2 years ago
- Baby monitoring device that uses MIT's "Eulerian Video Magnification" to detect heart beat based on facial colour changes.☆15Feb 26, 2017Updated 9 years ago
- Lightweight C++ cross-platform named shared memory interface.☆19Dec 20, 2011Updated 14 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- LLM plugin to generate plugins for LLM☆13Dec 30, 2024Updated last year
- Turning SymPy expressions into JAX functions☆45Mar 17, 2021Updated 5 years ago
- ☆50May 30, 2021Updated 5 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆20Oct 9, 2022Updated 3 years ago
- UCI Datasets for Haskell☆11Mar 17, 2017Updated 9 years ago
- JMP is a Mixed Precision library for JAX.☆213Jul 8, 2026Updated 2 months ago
- (Intellij) decompiles jar library and attaches sources to current project☆18Nov 13, 2017Updated 8 years ago
- JAX implementation of Graph Attention Networks☆13Jan 29, 2022Updated 4 years ago
- A very simple variant of adversarial training that yields excellent results on MNIST☆12Mar 19, 2016Updated 10 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Jan 16, 2022Updated 4 years ago
- ☆17Feb 21, 2026Updated 6 months ago
- ✨💯Gaussian Mixture Models in JAX 💯✨☆26Aug 7, 2026Updated last month
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- A constant-memory approximation of set membership☆25Jan 25, 2025Updated last year
- Linear Prediction Coefficients estimation from mel-spectrogram implemented in Python based on Levinson-Durbin algorithm.☆72Mar 19, 2021Updated 5 years ago
- PyTorch implementation of GLOM☆23Apr 1, 2022Updated 4 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19Aug 12, 2026Updated 3 weeks ago
- bfloat16 dtype for numpy☆20Sep 25, 2023Updated 2 years ago
- Distributed k-nearest Neighbors using Locality Sensitive Hashing and SYCL☆10Jul 26, 2026Updated last month