kasper0406 / stablehlo-coremlLinks
Convert StableHLO models into Apple Core ML format
☆19Updated 3 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:
- ☆337Updated this week
- ☆53Updated last year
- Backward compatible ML compute opset inspired by HLO/MHLO☆559Updated this week
- Stores documents and resources used by the OpenXLA developer community☆131Updated last year
- jax-triton contains integrations between JAX and OpenAI Triton☆432Updated 3 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆585Updated this week
- MLIR-based partitioning system☆144Updated this week
- High-Performance SGEMM on CUDA devices☆109Updated 9 months ago
- ☆180Updated last year
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆101Updated last week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆394Updated this week
- JAX-Toolbox☆359Updated this week
- Fastest kernels written from scratch☆386Updated last month
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆124Updated last month
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆539Updated 2 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆305Updated last week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆360Updated this week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆580Updated 2 months ago
- ☆246Updated this week
- Fast low-bit matmul kernels in Triton☆392Updated 2 weeks ago
- Ahead of Time (AOT) Triton Math Library☆81Updated this week
- IREE's PyTorch Frontend, based on Torch Dynamo.☆99Updated this week
- ☆190Updated 2 weeks ago
- TORCH_LOGS parser for PT2☆63Updated last month
- An open-source efficient deep learning framework/compiler, written in python.☆733Updated 2 months ago
- A library to analyze PyTorch traces.☆421Updated last week
- extensible collectives library in triton☆90Updated 7 months ago
- ☆51Updated last week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆55Updated 3 weeks ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆147Updated 2 years ago