iree-org/iree-jax external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

iree-org/iree-jax

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/iree-org/iree-jax)

Close

iree-org / iree-jaxView on GitHubMore

• External links
• Readme badge

☆56Aug 9, 2024Updated 2 years ago

Alternatives and similar repositories for iree-jax

Users that are interested in iree-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

• iree-org / iree-torch

  View on GitHub
  Torch Frontend for IREE
  ☆26Dec 21, 2023Updated 2 years ago
• iree-org / iree-experimental

  View on GitHub
  Experiments and prototypes associated with IREE or MLIR
  ☆57Aug 9, 2024Updated 2 years ago
• NeilGirdhar / tjax

  View on GitHub
  Tools for JAX
  ☆51Aug 28, 2026Updated 3 weeks ago
• PierrickPochelu / JaxDecompiler

  View on GitHub
  Jax Decompiler
  ☆24Apr 22, 2025Updated last year
• google / CommonLoopUtils

  View on GitHub
  CLU lets you write beautiful training loops in JAX.
  ☆371Jul 7, 2026Updated 2 months ago
• Wordpress hosting with auto-scaling - Free Trial Offer • Ad
  Fully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
• iree-org / iree-llvm-sandbox

  View on GitHub
  A sandbox for quick iteration and experimentation on projects related to IREE, MLIR, and LLVM
  ☆61Apr 13, 2026Updated 5 months ago
• pengmai / lagrad

  View on GitHub
  An MLIR-based source-to-source automatic differentiation system.
  ☆15Mar 30, 2023Updated 3 years ago
• j-towns / scanagram

  View on GitHub
  Tidy autoregressive inference in JAX
  ☆15Sep 1, 2025Updated last year
• iree-org / iree

  View on GitHub
  A retargetable MLIR-based machine learning compiler and runtime toolkit.
  ☆3,950Updated this week
• mattjj / ode-diff-notes

  View on GitHub
  ☆12Oct 8, 2020Updated 5 years ago
• lucidrains / jax2torch

  View on GitHub
  Use Jax functions in Pytorch
  ☆263Jul 1, 2023Updated 3 years ago
• nod-ai / AMD-SHARK-ModelDev

  View on GitHub
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆100Jul 2, 2026Updated 2 months ago
• nestordemeure / jochastic

  View on GitHub
  A JAX implementation of stochastic addition.
  ☆13Aug 15, 2022Updated 4 years ago
• cgarciae / dyn_plot

  View on GitHub
  ☆13Jul 12, 2024Updated 2 years ago
• AI Agents on DigitalOcean Gradient AI Platform • Ad
  Build production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
• breandan / cstk

  View on GitHub
  🔍 Code Search Tools & Experiments
  ☆13Sep 18, 2026Updated last week
• mmperf / mmperf

  View on GitHub
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆138Sep 25, 2023Updated 3 years ago
• google-deepmind / dm_aux

  View on GitHub
  ☆68Aug 21, 2024Updated 2 years ago
• MingliSun / MLIR-TVM

  View on GitHub
  ☆13Nov 25, 2019Updated 6 years ago
• kasper0406 / stablehlo-coreml

  View on GitHub
  Convert StableHLO models into Apple Core ML format
  ☆22Sep 12, 2026Updated 2 weeks ago
• iml130 / iree-template-cpp

  View on GitHub
  IREE C++ Template
  ☆17Jul 30, 2024Updated 2 years ago
• jax-ml / jax-ai-stack

  View on GitHub
  ☆311Updated this week
• ericjang / pt-jax

  View on GitHub
  Path Tracing in JAX
  ☆75Dec 15, 2019Updated 6 years ago
• google / jaxonnxruntime

  View on GitHub
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆138Sep 15, 2026Updated last week
• Wordpress hosting with auto-scaling - Free Trial Offer • Ad
  Fully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
• JuliaLLVM / MLIR.jl

  View on GitHub
  Julia library to manipulate MLIR dialects.
  ☆68Aug 14, 2026Updated last month
• nod-ai / pandas-mlir

  View on GitHub
  Bridging Pandas and MLIR ecosystems
  ☆18Nov 17, 2023Updated 2 years ago
• dfm / extending-jax

  View on GitHub
  Extending JAX with custom C++ and CUDA code
  ☆403Aug 18, 2024Updated 2 years ago
• lucidrains / mlp-gpt-jax

  View on GitHub
  A GPT, made only of MLPs, in Jax
  ☆59Jun 23, 2021Updated 5 years ago
• MPACT-ORG / mpact-compiler

  View on GitHub
  Retargetable ML compilers for the twenty-first century!
  ☆13Apr 22, 2025Updated last year
• ROCm / rocMLIR

  View on GitHub
  ☆193Sep 15, 2026Updated last week
• mila-iqia / torch_jax_interop

  View on GitHub
  Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
  ☆38Jan 8, 2026Updated 8 months ago
• google / aqt

  View on GitHub
  ☆362Sep 16, 2026Updated last week
• tachukao / ocaml-mlir

  View on GitHub
  OCaml Bindings to MLIR
  ☆16Dec 11, 2020Updated 5 years ago
• Wordpress hosting with auto-scaling - Free Trial Offer • Ad
  Fully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
• n2cholas / progan-flax

  View on GitHub
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12May 24, 2021Updated 5 years ago
• NodLabs / mlir-examples

  View on GitHub
  a simple end to end example of taking a ML graph (TF2 / PyTorch) and running it on a device [cpu, gpu]
  ☆37Jan 20, 2021Updated 5 years ago
• kingoflolz / swarm-jax

  View on GitHub
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆241May 12, 2023Updated 3 years ago
• lucidrains / local-attention-flax

  View on GitHub
  Local Attention - Flax module for Jax
  ☆22May 26, 2021Updated 5 years ago
• brentyi / jax_dataclasses

  View on GitHub
  Pytrees + dataclasses ❤️
  ☆76Dec 19, 2025Updated 9 months ago
• brentyi / jaxls

  View on GitHub
  Sparse nonlinear least squares in JAX
  ☆302Jun 22, 2026Updated 3 months ago
• f0uriest / quadax

  View on GitHub
  Numerical quadrature with JAX
  ☆91Updated this week