kasper0406 / stablehlo-coremlLinks
Convert StableHLO models into Apple Core ML format
☆19Updated 2 weeks ago
Alternatives and similar repositories for stablehlo-coreml
Users that are interested in stablehlo-coreml are comparing it to the libraries listed below
Sorting:
- ☆52Updated last year
- ☆324Updated last week
- Backward compatible ML compute opset inspired by HLO/MHLO☆517Updated this week
- MLIR-based partitioning system☆117Updated this week
- Stores documents and resources used by the OpenXLA developer community☆126Updated last year
- jax-triton contains integrations between JAX and OpenAI Triton☆412Updated last month
- ☆171Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆119Updated last week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆286Updated 2 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆212Updated this week
- High-Performance SGEMM on CUDA devices☆98Updated 6 months ago
- Ahead of Time (AOT) Triton Math Library☆75Updated last week
- extensible collectives library in triton☆88Updated 4 months ago
- ☆187Updated last week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆346Updated this week
- Fast low-bit matmul kernels in Triton☆339Updated last week
- JAX-Toolbox☆329Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆523Updated last week
- ☆49Updated 2 months ago
- ☆11Updated last year
- Fastest kernels written from scratch☆311Updated 4 months ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆52Updated last month
- FlashAttention (Metal Port)☆514Updated 10 months ago
- ☆420Updated this week
- TORCH_LOGS parser for PT2☆48Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆414Updated this week
- ☆227Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆567Updated this week
- An experimental CPU backend for Triton☆139Updated 2 months ago
- IREE's PyTorch Frontend, based on Torch Dynamo.☆94Updated this week