Alternatives and similar repositories for asst4-trainium

Users that are interested in asst4-trainium are comparing it to the libraries listed below

• aws-neuron / nki-samples
  ☆35Updated last week
• cchan / tccl
  extensible collectives library in triton
  ☆87Updated 2 months ago
• triton-lang / kernels
  ☆80Updated 6 months ago
• stanford-cs149 / cs149gpt
  ☆72Updated last year
• Deep-Learning-Profiling-Tools / triton-viz
  ☆215Updated this week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆133Updated last year
• Jokeren / triton-samples
  ☆28Updated 4 months ago
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆119Updated this week
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆43Updated 2 months ago
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆153Updated this week
• openxla / shardy
  MLIR-based partitioning system
  ☆86Updated this week
• stanford-cs149 / asst3
  Stanford CS149 -- Assignment 3
  ☆27Updated 7 months ago
• moritztng / grayskull-attention
  Attention in SRAM on Tenstorrent Grayskull
  ☆35Updated 10 months ago
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆45Updated 2 weeks ago
• Ying1123 / awesome-neural-symbolic
  A list of awesome neural symbolic papers.
  ☆47Updated 2 years ago
• ademeure / cuda-side-boost
  ☆19Updated last month
• gpu-mode / ring-attention
  ring-attention experiments
  ☆145Updated 7 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆88Updated last week
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆127Updated this week
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆125Updated 2 months ago
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆83Updated last year
• HazyResearch / Megakernels
  kernels, of the mega variety
  ☆329Updated this week
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆271Updated last week
• open-lm-engine / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆48Updated this week
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆229Updated 4 months ago
• UofT-EcoSystem / BPPSA-open
  The (open-source part of) code to reproduce "BPPSA: Scaling Back-propagation by Parallel Scan Algorithm".
  ☆12Updated 3 years ago
• uccl-project / uccl
  Ultra | Ultimate | Unified CCL
  ☆102Updated this week
• simveit / effective_transpose
  Effective transpose on Hopper GPU
  ☆20Updated last month
• awslabs / slapo
  A schedule language for large model training
  ☆148Updated 11 months ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆41Updated 10 months ago