kasper0406 / stablehlo-coremlLinks
Convert StableHLO models into Apple Core ML format
☆19Updated last month
Alternatives and similar repositories for stablehlo-coreml
Users that are interested in stablehlo-coreml are comparing it to the libraries listed below
Sorting:
- ☆328Updated last week
- ☆52Updated last year
- Backward compatible ML compute opset inspired by HLO/MHLO☆529Updated last week
- Stores documents and resources used by the OpenXLA developer community☆128Updated last year
- jax-triton contains integrations between JAX and OpenAI Triton☆415Updated 2 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆292Updated 3 weeks ago
- MLIR-based partitioning system☆125Updated this week
- High-Performance SGEMM on CUDA devices☆99Updated 7 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆274Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆122Updated last month
- Fast low-bit matmul kernels in Triton☆356Updated last week
- ☆174Updated last year
- ☆234Updated 2 weeks ago
- JAX-Toolbox☆332Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆51Updated this week
- ☆50Updated 3 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆138Updated last year
- TORCH_LOGS parser for PT2☆57Updated last week
- ☆145Updated 3 weeks ago
- ☆188Updated this week
- extensible collectives library in triton☆88Updated 5 months ago
- Ahead of Time (AOT) Triton Math Library☆76Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆351Updated this week
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆101Updated 2 weeks ago
- Minimal yet performant LLM examples in pure JAX☆150Updated last week
- FlashAttention (Metal Port)☆526Updated 11 months ago
- IREE's PyTorch Frontend, based on Torch Dynamo.☆95Updated this week
- ☆115Updated 5 months ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆530Updated this week
- Fastest kernels written from scratch☆318Updated 5 months ago