A retargetable MLIR-based machine learning compiler and runtime toolkit.

Functional programming language for signal processing and sound synthesis

Digital Audio Workstation with Python; VST instruments/effects, parameter automation, FAUST, JAX, Warp Markers, and JUCE processors

PySCF on IPU

[CVPR 2024] Confronting Ambiguity in 6D Object Pose Estimation via Score-Based Diffusion on SE(3)

distill large scale web page text

PjRt plugin and Python APIs for MPMD workflows in Jax

A MNIST model is defined and trained in Jax, then converted to TensorRT for inference.

JAX interface for XSPEC spectral models.

XLA Launcher is a high-performance, lightweight C++ library designed to provide a simple interface for loading and executing computation graphs represented in the StableHLO format.

SyntraLine++ is a compiler-based domain-specific language (DSL) for defining, validating, and executing machine learning pipelines with PyTorch and JAX backends with C++ architectural foundations.

Event-based runtime for gang-scheduling multi-GPU operations across sharded arrays

JAX Foreign Function Interface experiments