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
• PierrickPochelu / JaxDecompiler

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

  View on GitHub
  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
• microsoft / tensorflow-rematerialization

  View on GitHub
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Dec 20, 2019Updated 6 years ago
• 1-Click AI Models by DigitalOcean Gradient • Ad
  Deploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
• google / CommonLoopUtils

  View on GitHub
  CLU lets you write beautiful training loops in JAX.
  ☆369Jul 7, 2026Updated 2 months ago
• 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 4 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,919Updated 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
  ☆262Jul 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
• Managed Kubernetes at scale on DigitalOcean • Ad
  DigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
• cgarciae / dyn_plot

  View on GitHub
  ☆13Jul 12, 2024Updated 2 years ago
• breandan / cstk

  View on GitHub
  🔍 Code Search Tools & Experiments
  ☆13Updated this week
• wsmoses / Enzyme-GPU-Tests

  View on GitHub
  This repo contains the benchmarks for Enzyme on GPU's
  ☆11Aug 11, 2026Updated 3 weeks ago
• libxsmm / tpp-mlir

  View on GitHub
  TPP experimentation on MLIR for linear algebra
  ☆162Updated this week
• mmperf / mmperf

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

  View on GitHub
  ☆68Aug 21, 2024Updated 2 years ago
• kasper0406 / stablehlo-coreml

  View on GitHub
  Convert StableHLO models into Apple Core ML format
  ☆22Updated this week
• iml130 / iree-template-cpp

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

  View on GitHub
  ☆311Aug 31, 2026Updated last week
• 1-Click AI Models by DigitalOcean Gradient • Ad
  Deploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
• ericjang / pt-jax

  View on GitHub
  Path Tracing in JAX
  ☆75Dec 15, 2019Updated 6 years ago
• jax-ml / oryx

  View on GitHub
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆328Aug 6, 2026Updated last month
• google / jaxloop

  View on GitHub
  ☆21Jul 7, 2026Updated 2 months ago
• google / jaxonnxruntime

  View on GitHub
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆137Jul 6, 2026Updated 2 months ago
• cgarciae / ciclo

  View on GitHub
  A functional training loops library for JAX
  ☆87Feb 13, 2024Updated 2 years ago
• 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
  ☆404Aug 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
• ROCm / rocMLIR

  View on GitHub
  ☆192Updated this week
• Deploy to Railway using AI coding agents - Free Credits Offer • Ad
  Use Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
• 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 7 months ago
• 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
• darrenjw / fp-ssc-course

  View on GitHub
  An introduction to functional programming for scalable statistical computing
  ☆80Jul 2, 2024Updated 2 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 8 months ago