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:
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Jax Decompiler☆24Apr 22, 2025Updated last year
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated last year
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- Templates for commonly used GitHub actions steps☆13Dec 13, 2024Updated last year
- 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.
- An interface between ffmpeg and Lua/Torch☆20Jul 16, 2017Updated 9 years ago
- Zero Shot Molecular Generation via Similarity Kernels☆31Aug 30, 2026Updated 3 weeks ago
- CUDA implementations of MACE models☆22Aug 19, 2025Updated last year
- Docker container for ocropus3 OCR system☆13Aug 19, 2018Updated 8 years ago
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆36Apr 25, 2024Updated 2 years ago
- ☆29Jan 12, 2022Updated 4 years ago
- ☆10Dec 21, 2024Updated last year
- ☆18Dec 2, 2024Updated last year
- Torch bindings for FFmpeg (reading videos only)☆26Jul 13, 2016Updated 10 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.
- (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX☆33Sep 15, 2026Updated last week
- Graph algorithms for machine learning frameworks☆30Mar 22, 2023Updated 3 years ago
- A Fluid Simulator on GPU☆34Jul 1, 2021Updated 5 years ago
- fun visualization scripts☆10Updated this week
- PySCF on IPU☆44Mar 12, 2024Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆44May 9, 2026Updated 4 months ago
- JAX Arrays for human consumption☆113Oct 23, 2025Updated 11 months ago
- LuaJIT General Purpose Routines☆15Feb 11, 2021Updated 5 years ago
- A webapp for monitoring GPU machines, written in Vue.js and Flask.☆23Nov 21, 2021Updated 4 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 simple architecture to one complete environment of a IoT System.☆21Sep 25, 2020Updated 5 years ago
- Parallel hyperparameter tuning with JAX☆40Jul 18, 2026Updated 2 months ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 6 years ago
- Python tabular data package.☆17Oct 23, 2013Updated 12 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆361Sep 4, 2026Updated 2 weeks ago
- Generic floating-point types in Python☆20Updated this week
- Train very large language models in Jax.☆208Oct 21, 2023Updated 2 years ago
- A Python library for machine-learning and feedback loops on streaming data☆64Aug 15, 2026Updated last month
- Abelian sandpiles☆16Nov 16, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Collection of Processing sketches to accompany Jared S Tarbell's Generative Spaces presentation.☆32Oct 5, 2016Updated 9 years ago
- ☆18Mar 20, 2022Updated 4 years ago
- MiraPy: A Python package for Deep Learning in Astronomy☆45May 26, 2020Updated 6 years ago
- A Symbolic Emulator for Shuffle Synthesis on the NVIDIA PTX Code☆16Mar 19, 2023Updated 3 years ago
- Turning Python into a probabilistic programming language☆19Sep 2, 2014Updated 12 years ago
- Entity linker for the newspaper collection of the National Library of the Netherlands. Links named entity mentions to DBpedia description…☆11Dec 8, 2022Updated 3 years ago
- ExploitGen is a template-augmented Exploit Code generation method based on CodeBERT, which accepted in JSS.☆11Feb 9, 2024Updated 2 years ago