kasper0406 / stablehlo-coreml
Convert StableHLO models into Apple Core ML format
☆16Updated 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
- ☆51Updated 7 months ago
- ☆290Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆109Updated last month
- Stores documents and resources used by the OpenXLA developer community☆117Updated 7 months ago
- MLIR-based partitioning system☆73Updated this week
- extensible collectives library in triton☆84Updated 6 months ago
- High-Performance SGEMM on CUDA devices☆86Updated 2 months ago
- Fast low-bit matmul kernels in Triton☆267Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆386Updated last week
- ☆21Updated 3 weeks ago
- ☆203Updated 2 months ago
- ☆191Updated this week
- ModernBERT model optimized for Apple Neural Engine.☆23Updated 2 months ago
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆255Updated this week
- Applied AI experiments and examples for PyTorch☆250Updated this week
- Experiment of using Tangent to autodiff triton☆78Updated last year
- TORCH_LOGS parser for PT2☆35Updated last week
- seqax = sequence modeling + JAX☆150Updated last week
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆237Updated last week
- FlashAttention (Metal Port)☆457Updated 6 months ago
- Backward compatible ML compute opset inspired by HLO/MHLO☆457Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆34Updated this week
- JAX-Toolbox☆289Updated this week
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆221Updated this week
- Fastest kernels written from scratch☆199Updated 2 weeks ago
- ☆162Updated 9 months ago
- ☆197Updated this week
- ☆22Updated last week
- Ahead of Time (AOT) Triton Math Library☆55Updated this week