iree-org / iree-jaxLinks
☆52Updated 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:
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆121Updated 3 weeks ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆291Updated 2 weeks ago
- Convert StableHLO models into Apple Core ML format☆19Updated last month
- MLIR-based partitioning system☆123Updated last week
- jax-triton contains integrations between JAX and OpenAI Triton☆414Updated 2 months ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- TORCH_LOGS parser for PT2☆57Updated this week
- High-Performance SGEMM on CUDA devices☆97Updated 7 months ago
- Stores documents and resources used by the OpenXLA developer community☆127Updated last year
- Torch Frontend for IREE☆25Updated last year
- An MLIR-based source-to-source automatic differentiation system.☆14Updated 2 years ago
- Experiments and prototypes associated with IREE or MLIR☆55Updated last year
- ☆325Updated 3 weeks ago
- ☆28Updated 7 months ago
- JMP is a Mixed Precision library for JAX.☆208Updated 6 months ago
- ☆50Updated 3 months ago
- Experiment of using Tangent to autodiff triton☆80Updated last year
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆93Updated last week
- POC work on MLIR backend☆58Updated last year
- A functional training loops library for JAX☆88Updated last year
- ☆60Updated 2 weeks ago
- OpTree: Optimized PyTree Utilities☆190Updated last week
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- Tensor Compute Primitives: Mid-level Intermediate Representation for Machine Learning Programs☆35Updated 7 months ago
- Compiler with automatic differentiation☆48Updated last year
- ☆105Updated last year
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆109Updated last year
- ☆16Updated 11 months ago
- Python interface for MLIR - the Multi-Level Intermediate Representation☆264Updated 9 months ago
- Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …☆106Updated 7 months ago