kasper0406 / stablehlo-coremlLinks
Convert StableHLO models into Apple Core ML format
☆19Updated 2 months ago
Alternatives and similar repositories for stablehlo-coreml
Users that are interested in stablehlo-coreml are comparing it to the libraries listed below
Sorting:
- ☆322Updated 3 weeks ago
- Backward compatible ML compute opset inspired by HLO/MHLO☆501Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆407Updated 3 weeks ago
- Stores documents and resources used by the OpenXLA developer community☆126Updated 11 months ago
- ☆52Updated 11 months ago
- ☆169Updated last year
- High-Performance SGEMM on CUDA devices☆97Updated 5 months ago
- MLIR-based partitioning system☆105Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆281Updated last week
- Fast low-bit matmul kernels in Triton☆330Updated last week
- extensible collectives library in triton☆87Updated 3 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆116Updated this week
- ☆225Updated last week
- ☆186Updated last month
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆187Updated this week
- Fastest kernels written from scratch☆292Updated 3 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 11 months ago
- Composable Function Transformations in Python with Mojo/MAX acceleration☆267Updated last week
- Ahead of Time (AOT) Triton Math Library☆70Updated last week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆343Updated this week
- ☆214Updated 5 months ago
- ☆48Updated last month
- An open-source efficient deep learning framework/compiler, written in python.☆710Updated last week
- JAX-Toolbox☆323Updated this week
- The simplest but fast implementation of matrix multiplication in CUDA.☆37Updated 11 months ago
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆67Updated last week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆514Updated last week
- An experimental CPU backend for Triton☆135Updated last month
- Applied AI experiments and examples for PyTorch☆286Updated last month
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆560Updated this week