hyhieu / easy_pybind

Related projects ⓘ

Alternatives and complementary repositories for easy_pybind

• vdesai2014 / inference-optimization-blog-post
  ☆82Updated 8 months ago
• apple / ml-cross-entropy
  ☆121Updated this week
• cloneofsimo / min-fsdp
  ☆73Updated 4 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆133Updated 9 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆76Updated last week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆104Updated last month
• unixpickle / learn-ptx
  Learning about CUDA by writing PTX code.
  ☆28Updated 8 months ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆121Updated 3 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 6 months ago
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆112Updated 2 months ago
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆90Updated 3 months ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆97Updated last month
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆79Updated 4 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆163Updated 3 months ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆134Updated 2 months ago
• pytorch / torchcodec
  PyTorch video decoding
  ☆79Updated this week
• cloneofsimo / efae
  ☆21Updated 5 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆61Updated 11 months ago
• mag- / gpu_benchmark
  Gpu benchmark
  ☆43Updated last month
• cloneofsimo / scaling-guide
  WIP
  ☆89Updated 3 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆211Updated 3 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆152Updated this week
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆191Updated 9 months ago
• ClashLuke / HeavyBall
  Efficient optimizers
  ☆79Updated this week
• CLAIRE-Labo / flash_attention
  A basic pure pytorch implementation of flash attention
  ☆16Updated 3 weeks ago
• yilundu / ired_code_release
  ☆46Updated 5 months ago