A stateful pytree library for training neural networks.
☆22Aug 31, 2025Updated 11 months ago
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
- Tools for JAX☆50Aug 11, 2026Updated last week
- Implementation of a Tensorflow XLA rematerialization pass☆15Dec 20, 2019Updated 6 years ago
- Convenient functions for working with panel (longitudinal) data in Julia☆14Jun 24, 2026Updated last month
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆72Jul 24, 2022Updated 4 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Matlab Mex implementation☆11May 2, 2016Updated 10 years ago
- Butterfly Reader for Hacker News, built with Flutter☆16Oct 27, 2019Updated 6 years ago
- 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
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- [DEPRECATED] A babel plugin that automatically assigns constructor parameters to instance properties.☆17Aug 11, 2018Updated 8 years ago
- ☆12Jul 6, 2023Updated 3 years ago
- nashorn websockets demo☆16Oct 2, 2013Updated 12 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- A simple systems monitoring dashboard which displays process information in realtime similar to what you would see when you run the unix …☆13Jun 5, 2017Updated 9 years ago
- ☆11May 12, 2021Updated 5 years ago
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 3 years ago
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- A Distributed Resilient Scheme Interpreter☆16Jul 9, 2013Updated 13 years ago
- Lightweight C++ cross-platform named shared memory interface.☆19Dec 20, 2011Updated 14 years ago
- ☆19Dec 9, 2019Updated 6 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
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Python library to implement bank api integration☆17Jan 29, 2026Updated 6 months ago
- Transformer from scratch with einsum method☆11Jul 8, 2021Updated 5 years ago
- The basic idea is to utilize a U-Net, modified perceptual loss (pearson instead of MAE), learned basis functions, and "mean teacher" tra…☆42Jan 8, 2019Updated 7 years ago
- JMP is a Mixed Precision library for JAX.☆213Jul 8, 2026Updated last month
- Companion repository to "Prompt Compression and Contrastive Conditioning for Controllability and Toxicity Reduction in Language Models"☆14May 31, 2023Updated 3 years ago
- ☆14Feb 19, 2021Updated 5 years 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
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Jan 16, 2022Updated 4 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆17Feb 21, 2026Updated 5 months ago
- Some tips on using stable diffusion inpainting with diffusers☆14Jul 19, 2023Updated 3 years ago
- ✨💯Gaussian Mixture Models in JAX 💯✨☆26Aug 7, 2026Updated last week
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- PyTorch implementation of GLOM☆23Apr 1, 2022Updated 4 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19Updated this week
- bfloat16 dtype for numpy☆20Sep 25, 2023Updated 2 years ago