kasper0406 / stablehlo-coreml

Related projects ⓘ

Alternatives and complementary repositories for stablehlo-coreml

• Deep-Learning-Profiling-Tools / triton-viz
  ☆152Updated this week
• google / aqt
  ☆267Updated last week
• apple / ml-recurrent-drafter
  ☆99Updated last month
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆166Updated 3 weeks ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆412Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆71Updated last month
• philipturner / metal-flash-attention
  FlashAttention (Metal Port)
  ☆386Updated last month
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆211Updated 3 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆270Updated this week
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆143Updated this week
• albanD / subclass_zoo
  ☆148Updated 5 months ago
• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆107Updated 3 months ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆307Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆343Updated 3 weeks ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆134Updated 2 months ago
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆457Updated last week
• ml-explore / mlx-c
  C API for MLX
  ☆77Updated this week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆143Updated this week
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆245Updated this week
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆56Updated last week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆107Updated last year
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆483Updated 3 weeks ago
• rwitten / HighPerfLLMs2024
  ☆224Updated 4 months ago
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆35Updated 6 months ago
• openxla / shardy
  MLIR-based partitioning system
  ☆40Updated this week
• google / praxis
  ☆178Updated last week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆81Updated this week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆215Updated this week
• intel / intel-extension-for-openxla
  ☆39Updated 2 months ago
• iree-org / iree-jax
  ☆48Updated 3 months ago