thuml / learn_torch.compile

Related projects: ⓘ

• tridao / cutlass_quant
  ☆16Updated this week
• megvii-research / IntLLaMA
  IntLLaMA: A fast and light quantization solution for LLaMA
  ☆18Updated last year
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆10Updated last month
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆58Updated 6 months ago
• tridao / flash-attention-wheels
  ☆38Updated 9 months ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆20Updated 5 months ago
• chips-compilers-mlsys-21 / chips-compilers-mlsys-21.github.io
  ☆11Updated 3 years ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆50Updated 3 months ago
• microsoft / DeepSpeed-Kernels
  ☆50Updated 3 months ago
• BBuf / flash-rwkv
  ☆28Updated 3 months ago
• Raincleared-Song / sparse_gpu_operator
  GPU operators for sparse tensor operations
  ☆27Updated 6 months ago
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆35Updated 4 months ago
• vedantroy / gpu_kernels
  ☆22Updated 8 months ago
• ptillet / triton-llvm-releases
  ☆20Updated 11 months ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆25Updated last month
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆73Updated 3 months ago
• LeiWang1999 / Stream-k.tvm
  ☆14Updated last week
• Aleph-Alpha / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆23Updated 9 months ago
• Ying1123 / awesome-neural-symbolic
  A list of awesome neural symbolic papers.
  ☆37Updated 2 years ago
• yuzhenmao / IceFormer
  Implementation of IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).
  ☆18Updated 3 months ago
• enp1s0 / ozIMMU
  FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme
  ☆44Updated 2 weeks ago
• rayleizhu / vllm-ra
  [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts
  ☆29Updated 6 months ago
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆13Updated 2 years ago
• openxla / openxla-nvgpu
  ☆48Updated 6 months ago
• buttercutter / Mamba_SSM
  A simple implementation of [Mamba: Linear-Time Sequence Modeling with Selective State Spaces](https://arxiv.org/abs/2312.00752)
  ☆19Updated 7 months ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• feifeibear / PSTensor
  PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.
  ☆9Updated 2 years ago
• stanford-futuredata / stk
  ☆83Updated 3 weeks ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated 11 months ago