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:
- Jax Decompiler☆16Apr 22, 2025Updated last year
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 10 months ago
- A library for unit scaling in PyTorch☆135Jul 11, 2025Updated last year
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- Zero Shot Molecular Generation via Similarity Kernels☆29Aug 27, 2025Updated 10 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆29Jan 12, 2022Updated 4 years ago
- ☆10Dec 21, 2024Updated last year
- ☆18Dec 2, 2024Updated last year
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- fun visualization scripts☆10Mar 1, 2026Updated 4 months ago
- MESS: Modern Electronic Structure Simulations☆43May 9, 2026Updated 2 months ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Jul 29, 2024Updated last year
- PubNative Dockerfiles library☆16Jan 15, 2026Updated 6 months ago
- LuaJIT General Purpose Routines☆15Feb 11, 2021Updated 5 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.
- Parallel hyperparameter tuning with JAX☆39Updated this week
- Automatic Differentiation for high-performance stencil loops☆13Mar 25, 2021Updated 5 years ago
- Command-line corpus tools☆12May 15, 2017Updated 9 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- A quantum chemisty library in jax.☆44Feb 14, 2026Updated 5 months ago
- ☆11Sep 20, 2016Updated 9 years ago
- 🚀 Sliding Window Attention Training for Efficient Large Language Models☆19Jun 7, 2026Updated last month
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆351Jun 30, 2026Updated 3 weeks ago
- Generic floating-point types in Python☆20Apr 18, 2026Updated 3 months 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.
- Train very large language models in Jax.☆208Oct 21, 2023Updated 2 years ago
- Triton kernels for dynamic causal short convolutions.☆24Jun 4, 2026Updated last month
- Comparing ndarray abstractions for complex workflows☆15Sep 22, 2015Updated 10 years ago
- Abelian sandpiles☆16Nov 16, 2024Updated last year
- ☆18Mar 20, 2022Updated 4 years ago
- ☆14Dec 22, 2015Updated 10 years ago
- Turning Python into a probabilistic programming language☆19Sep 2, 2014Updated 11 years ago
- ☆12Dec 19, 2016Updated 9 years ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆53Sep 14, 2024Updated last year
- 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.
- A fast, safe, and portable implementation of HighwayHash for Rust☆18Jun 6, 2023Updated 3 years ago
- Talk for PyMNtos on code generation in SymPy☆18Sep 18, 2014Updated 11 years ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆15May 10, 2024Updated 2 years ago
- A tiny realtime Vulkan 3D engine with modern graphics, all implementations in the 📄ZeldaEngine.cpp file.☆14Jul 1, 2024Updated 2 years ago
- Generates unplayable music in the style of John Stump's "Faerie’s Aire and Death Waltz".☆12Nov 9, 2019Updated 6 years ago
- Dynamical Systems with JAX☆12Jun 3, 2026Updated last month
- A library & tools to evaluate predictive language models.☆66Aug 9, 2023Updated 2 years ago