Alternatives and similar repositories for digestai

Users that are interested in digestai are comparing it to the libraries listed below

• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆63Updated this week
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆42Updated 2 months ago
• masahi / torchscript-to-tvm
  ☆69Updated 2 years ago
• mlc-ai / mlc-python
  ☆33Updated this week
• iree-org / iree-nvgpu
  ☆50Updated last year
• justinchuby / model-explorer-onnx
  Visualize ONNX models with model-explorer
  ☆33Updated 2 months ago
• octoml / relax
  A fork of tvm/unity
  ☆14Updated last year
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆31Updated 3 months ago
• octoml / octoml-profile
  Home for OctoML PyTorch Profiler
  ☆113Updated 2 years ago
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆138Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆109Updated 10 months ago
• Jokeren / triton-samples
  ☆27Updated 4 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆85Updated this week
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆28Updated last year
• tlc-pack / tlcpack
  ☆24Updated last year
• ROCm / aiter
  AI Tensor Engine for ROCm
  ☆195Updated this week
• openxla / shardy
  MLIR-based partitioning system
  ☆82Updated this week
• onnx / neural-compressor
  Model compression for ONNX
  ☆92Updated 6 months ago
• gau-nernst / quantized-training
  Explore training for quantized models
  ☆18Updated 4 months ago
• mobiusml / gemlite
  Fast low-bit matmul kernels in Triton
  ☆301Updated this week
• intel / intel-extension-for-deepspeed
  Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…
  ☆61Updated 2 months ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆108Updated 8 months ago
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆100Updated this week
• deepspeedai / DeepSpeed-Kernels
  ☆69Updated last month
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆69Updated 3 years ago
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆43Updated this week
• triton-lang / kernels
  ☆79Updated 6 months ago
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆97Updated 3 weeks ago
• awslabs / slapo
  A schedule language for large model training
  ☆147Updated 11 months ago
• tlc-pack / TLCBench
  Benchmark scripts for TVM
  ☆74Updated 3 years ago