kasper0406 / stablehlo-coremlLinks
Convert StableHLO models into Apple Core ML format
☆19Updated last week
Alternatives and similar repositories for stablehlo-coreml
Users that are interested in stablehlo-coreml are comparing it to the libraries listed below
Sorting:
- ☆337Updated last week
- Backward compatible ML compute opset inspired by HLO/MHLO☆573Updated last week
- jax-triton contains integrations between JAX and OpenAI Triton☆436Updated this week
- ☆54Updated last year
- Stores documents and resources used by the OpenXLA developer community☆131Updated last year
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆640Updated this week
- MLIR-based partitioning system☆150Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆312Updated this week
- ☆184Updated last year
- High-Performance SGEMM on CUDA devices☆112Updated 10 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆125Updated 2 months ago
- IREE's PyTorch Frontend, based on Torch Dynamo.☆101Updated this week
- Fast low-bit matmul kernels in Triton☆401Updated last week
- An open-source efficient deep learning framework/compiler, written in python.☆736Updated 2 months ago
- extensible collectives library in triton☆91Updated 8 months ago
- Ahead of Time (AOT) Triton Math Library☆84Updated 2 weeks ago
- ☆256Updated this week
- Fast and Furious AMD Kernels☆298Updated this week
- Tokamax: A GPU and TPU kernel library.☆116Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆540Updated last week
- TORCH_LOGS parser for PT2☆65Updated 2 weeks ago
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆404Updated this week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆151Updated 2 years ago
- ☆51Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆583Updated 3 months ago
- FlashAttention (Metal Port)☆557Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆226Updated last year
- Machine Learning library for the emerging Mojo/Python ecosystem☆296Updated last week
- ☆190Updated last week
- Fastest kernels written from scratch☆400Updated 2 months ago