Alternatives and similar repositories for torchtrail:

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

• tenstorrent / tt-metal
  TT-NN operator library, and TT-Metalium low level kernel programming model.
  ☆601Updated this week
• salykova / sgemm.cu
  SGEMM that beats cuBLAS
  ☆68Updated last week
• cchan / tccl
  extensible collectives library in triton
  ☆77Updated 4 months ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆91Updated 8 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆74Updated last year
• Deep-Learning-Profiling-Tools / triton-viz
  ☆171Updated last week
• manishucsd / py-codegen
  ☆15Updated 4 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆136Updated 6 months ago
• moritztng / grayskull-attention
  Attention in SRAM on Tenstorrent Grayskull
  ☆31Updated 6 months ago
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆99Updated last week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆240Updated this week
• lianakoleva / no-libtorch-compile
  ☆21Updated 3 months ago
• neuralmagic / compressed-tensors
  A safetensors extension to efficiently store sparse quantized tensors on disk
  ☆66Updated this week
• tenstorrent / tt-mlir
  Tenstorrent MLIR compiler
  ☆86Updated this week
• IST-DASLab / SparseFinetuning
  Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry
  ☆40Updated last year
• gau-nernst / quantized-training
  Explore training for quantized models
  ☆13Updated 3 weeks ago
• vdesai2014 / inference-optimization-blog-post
  ☆85Updated 11 months ago
• tenstorrent / pytorch2.0_ttnn
  ⭐️ TTNN Compiler for PyTorch 2.0 ⭐️ It enables running PyTorch2.0 models on Tenstorrent hardware
  ☆29Updated this week
• UmerHA / triton_util
  Make triton easier
  ☆44Updated 7 months ago
• HanGuo97 / flute
  Fast Matrix Multiplications for Lookup Table-Quantized LLMs
  ☆221Updated last week
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆44Updated last month
• iree-org / iree-jax
  ☆49Updated 5 months ago
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆131Updated 2 months ago
• triton-lang / kernels
  ☆64Updated 2 months ago
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆64Updated 4 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated 3 months ago
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆62Updated this week
• iree-org / iree-nvgpu
  ☆48Updated 10 months ago
• mlc-ai / mlc-python
  ☆21Updated this week
• CentML / DeepView.Profile
  🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.
  ☆58Updated last week