☆55Aug 9, 2024Updated last year
Alternatives and similar repositories for iree-jax
Users that are interested in iree-jax are comparing it to the libraries listed below
Sorting:
- Torch Frontend for IREE☆25Dec 21, 2023Updated 2 years ago
- Experiments and prototypes associated with IREE or MLIR☆56Aug 9, 2024Updated last year
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆103Dec 22, 2025Updated 2 months ago
- An MLIR-based source-to-source automatic differentiation system.☆15Mar 30, 2023Updated 2 years ago
- IREE C++ Template☆17Jul 30, 2024Updated last year
- Conversions to MLIR EmitC☆134Dec 12, 2024Updated last year
- Retargetable ML compilers for the twenty-first century!☆13Apr 22, 2025Updated 10 months ago
- ☆12Oct 8, 2020Updated 5 years ago
- This repo contains the benchmarks for Enzyme on GPU's☆11Feb 22, 2026Updated 2 weeks ago
- Path Tracing in JAX☆73Dec 15, 2019Updated 6 years ago
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,621Updated this week
- Implementation of a Tensorflow XLA rematerialization pass☆15Dec 20, 2019Updated 6 years ago
- 🔍 Code Search Tools & Experiments☆12Mar 1, 2026Updated last week
- ☆13Jul 12, 2024Updated last year
- ☆21Feb 3, 2026Updated last month
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆133Mar 2, 2026Updated last week
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago
- Training AI for Super Smash Bros. Melee☆32Mar 27, 2025Updated 11 months ago
- ☆273Mar 2, 2026Updated last week
- TPP experimentation on MLIR for linear algebra☆146Feb 24, 2026Updated last week
- Plane Wave Density Functional Theory Code for the GPU☆12Jan 23, 2015Updated 11 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Feb 27, 2026Updated last week
- ☆13Nov 25, 2019Updated 6 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- Run WebAssembly in Elixir via Wasmtime☆13Jul 27, 2024Updated last year
- CoqTL allows users to write model transformations and prove engine/transformation correctness in Coq☆14Mar 11, 2022Updated 3 years ago
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆138Sep 25, 2023Updated 2 years ago
- ☆67Aug 21, 2024Updated last year
- Repo for solving arc problems with an Neural Cellular Automata☆23May 21, 2025Updated 9 months ago
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- Julia library to manipulate MLIR dialects.☆67Feb 24, 2026Updated last week
- Pytrees + dataclasses ❤️☆75Dec 19, 2025Updated 2 months ago
- Numerical quadrature with JAX☆78Mar 2, 2026Updated last week
- CS294 AI Systems Class Website☆17Apr 25, 2022Updated 3 years ago
- Image augmentation library for Jax☆41Apr 9, 2024Updated last year
- Nature's Cost Function (NCF). Finding paths of least action with gradient descent.☆18Mar 30, 2023Updated 2 years ago
- Computation offloading is a technique to circumvent device restrictions and bring novel, computationally-intensive applications to a hete…☆22Feb 4, 2026Updated last month
- Compiler with automatic differentiation☆49Oct 18, 2023Updated 2 years ago