thuml / learn_torch.compile
torch.compile artifacts for common deep learning models, can be used as a learning resource for torch.compile
☆13Updated 8 months ago
Related projects: ⓘ
- ☆16Updated this week
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- TensorRT LLM Benchmark Configuration☆10Updated last month
- Memory Optimizations for Deep Learning (ICML 2023)☆58Updated 6 months ago
- ☆38Updated 9 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆20Updated 5 months ago
- ☆11Updated 3 years ago
- Odysseus: Playground of LLM Sequence Parallelism☆50Updated 3 months ago
- ☆50Updated 3 months ago
- ☆28Updated 3 months ago
- GPU operators for sparse tensor operations☆27Updated 6 months ago
- GPTQ inference TVM kernel☆35Updated 4 months ago
- ☆22Updated 8 months ago
- ☆20Updated 11 months ago
- Hacks for PyTorch☆17Updated last year
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆25Updated last month
- Patch convolution to avoid large GPU memory usage of Conv2D☆73Updated 3 months ago
- ☆14Updated last week
- ☆23Updated 9 months ago
- A list of awesome neural symbolic papers.☆37Updated 2 years ago
- Implementation of IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).☆18Updated 3 months ago
- FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme☆44Updated 2 weeks ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆29Updated 6 months ago
- An external memory allocator example for PyTorch.☆13Updated 2 years ago
- ☆48Updated 6 months ago
- A simple implementation of [Mamba: Linear-Time Sequence Modeling with Selective State Spaces](https://arxiv.org/abs/2312.00752)☆19Updated 7 months ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆43Updated last year
- 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
- ☆83Updated 3 weeks ago
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated 11 months ago