1 Repo
Processes for translating JAX-based research programs into deployable binaries for hardware execution.
Distinct from Model Deployments: Focuses on the translation of JAX's specific functional representation into a deployable format, not cloud orchestration.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · JAX Model Deployments. Refine with filters or upvote what's useful.
IREE is an MLIR-based compiler toolchain and runtime designed to translate machine learning models from various frameworks into optimized binaries for execution across diverse hardware targets. It provides a unified pipeline to ingest models from PyTorch, TensorFlow, JAX, and ONNX, lowering them into a common intermediate representation for deployment on CPUs, GPUs, and bare-metal embedded systems. The project distinguishes itself through a bytecode virtual machine and a hardware abstraction layer that decouple high-level model logic from specific hardware instruction sets. It supports sophis
Translates high-performance JAX research programs into optimized, deployable formats for target hardware.