JAX for Graphcore IPU (experimental)
☆21Mar 12, 2024Updated 2 years ago
Alternatives and similar repositories for jax-experimental
Users that are interested in jax-experimental are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- MESS: Modern Electronic Structure Simulations☆20Sep 24, 2024Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Jax Decompiler☆16Apr 22, 2025Updated last year
- Flexibly track outputs and grad-outputs of torch.nn.Module.☆13Oct 6, 2023Updated 2 years ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Reusable DFT Grids for the Masses☆19Jun 11, 2026Updated 3 weeks ago
- Zero Shot Molecular Generation via Similarity Kernels☆29Aug 27, 2025Updated 10 months ago
- An interface between ffmpeg and Lua/Torch☆20Jul 16, 2017Updated 8 years ago
- ☆11Oct 13, 2023Updated 2 years ago
- ☆16Dec 31, 2021Updated 4 years ago
- Code snippets and reproductions from JustAByte☆48Apr 6, 2026Updated 2 months ago
- Docker container for ocropus3 OCR system☆12Aug 19, 2018Updated 7 years ago
- ☆10Dec 21, 2024Updated last year
- ☆18Dec 2, 2024Updated last year
- 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 modular system for machinable research code☆36Updated this week
- Torch bindings for FFmpeg (reading videos only)☆26Jul 13, 2016Updated 9 years ago
- PhoneGap NFC peer to peer demo☆22Jan 6, 2017Updated 9 years ago
- Generic floating-point types in Python☆17Apr 18, 2026Updated 2 months ago
- PySCF on IPU☆44Mar 12, 2024Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆43May 9, 2026Updated last month
- JAX Arrays for human consumption☆112Oct 23, 2025Updated 8 months ago
- A simple React component that handles file drag and drop.☆12Sep 3, 2017Updated 8 years ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Jul 29, 2024Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Scaling RLLib for generic simulation environments on Theta☆20Feb 16, 2023Updated 3 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 10 months ago
- hopfield☆30Oct 8, 2021Updated 4 years ago
- PubNative Dockerfiles library☆16Jan 15, 2026Updated 5 months ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- ☆11Sep 20, 2016Updated 9 years ago
- fun visualization scripts☆10Mar 1, 2026Updated 4 months ago
- Decoupled Neural Interfaces Using Synthetic Gradients - under develeopment☆11Jun 27, 2025Updated last year
- Command-line corpus tools☆12May 15, 2017Updated 9 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Repository of Quantum Datasets Publicly Available☆65Jun 19, 2025Updated last year
- An efficient method for the conversion from internal to Cartesian coordinates that utilizes the platform-agnostic JAX Python library.☆21Jun 12, 2024Updated 2 years ago
- Abelian sandpiles☆16Nov 16, 2024Updated last year
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆348Jun 23, 2026Updated last week
- Train very large language models in Jax.☆208Oct 21, 2023Updated 2 years ago
- ☆18Mar 20, 2022Updated 4 years ago
- ☆14Dec 22, 2015Updated 10 years ago