Alternatives and similar repositories for torchtrail

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

• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆107Updated 9 months ago
• ROCm / iris
  AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming
  ☆101Updated this week
• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆131Updated last year
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆147Updated 2 years ago
• cchan / tccl
  extensible collectives library in triton
  ☆90Updated 7 months ago
• tenstorrent / tt-mlir
  Tenstorrent MLIR compiler
  ☆206Updated this week
• google / aqt
  ☆335Updated last month
• tenstorrent / tt-metal
  TT-NN operator library, and TT-Metalium low level kernel programming model.
  ☆1,250Updated this week
• bertmaher / simplegemm
  ☆123Updated 2 weeks ago
• dropbox / gemlite
  Fast low-bit matmul kernels in Triton
  ☆388Updated last week
• KernelTuner / kernel_tuner
  Kernel Tuner
  ☆371Updated this week
• MekkCyber / CutlassAcademy
  A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS
  ☆240Updated 6 months ago
• tenstorrent / pytorch2.0_ttnn
  ⭐️ TTNN Compiler for PyTorch 2 ⭐️ Enables running PyTorch models on Tenstorrent hardware using eager or compile path
  ☆60Updated this week
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆384Updated last month
• moritztng / grayskull-attention
  Attention in SRAM on Tenstorrent Grayskull
  ☆38Updated last year
• Deep-Learning-Profiling-Tools / triton-viz
  ☆246Updated this week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆305Updated last week
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆99Updated this week
• meta-pytorch / BackendBench
  How to ensure correctness and ship LLM generated kernels in PyTorch
  ☆111Updated last week
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆107Updated last year
• anthonix / llm.c
  LLM training in simple, raw C/HIP for AMD GPUs
  ☆51Updated last year
• openxla / shardy
  MLIR-based partitioning system
  ☆144Updated this week
• vdesai2014 / inference-optimization-blog-post
  ☆89Updated last year
• albanD / subclass_zoo
  ☆180Updated last year
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆55Updated 3 weeks ago
• tenstorrent / tt-buda
  Tenstorrent TT-BUDA Repository
  ☆315Updated 7 months ago
• tenstorrent / tt-forge-fe
  The TT-Forge FE is a graph compiler designed to optimize and transform computational graphs for deep learning models, enhancing their per…
  ☆51Updated this week
• gpu-mode / reference-kernels
  Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!
  ☆116Updated this week
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆192Updated 2 years ago
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆287Updated 3 months ago