lianakoleva / no-libtorch-compile

Related projects ⓘ

Alternatives and complementary repositories for no-libtorch-compile

• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆44Updated 2 weeks ago
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 5 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆71Updated last month
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆16Updated 4 months ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 3 months ago
• ethansmith2000 / TransformerExperiments
  ☆18Updated last month
• haileyschoelkopf / triton-index
  See https://github.com/cuda-mode/triton-index/ instead!
  ☆11Updated 6 months ago
• cloneofsimo / min-fsdp
  ☆73Updated 4 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated 2 weeks ago
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆35Updated 4 months ago
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆14Updated 2 weeks ago
• aredden / torch-cublas-hgemm
  PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu
  ☆39Updated 2 months ago
• srush / drop7
  ☆18Updated 7 months ago
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆22Updated last week
• mgmalek / efficient_cross_entropy
  ☆77Updated 5 months ago
• Ryu1845 / hyena-jax
  Implementation of Hyena Hierarchy in JAX
  ☆10Updated last year
• cat-state / tinypar
  ☆20Updated last year
• Doraemonzzz / Awesome-Triton-Resources
  Awesome Triton Resources
  ☆18Updated last month
• google-deepmind / asyncdiloco
  ☆39Updated 10 months ago
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated last year
• mag- / gpu_benchmark
  Gpu benchmark
  ☆43Updated last month
• yixiaoer / tpu-training-example
  ☆13Updated 4 months ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆14Updated 9 months ago
• Aleph-Alpha / trigrams
  ☆43Updated 2 months ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆46Updated 11 months ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated last year
• Chillee / lit-llama
  Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code
  ☆10Updated last year
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆19Updated this week