facebookresearch / RLCompOpt
Learning Compiler Pass Orders using Coreset and Normalized Value Prediction. (ICML 2023)
☆18Updated last year
Alternatives and similar repositories for RLCompOpt:
Users that are interested in RLCompOpt are comparing it to the libraries listed below
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated this week
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 7 months ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated last year
- NASRec Weight Sharing Neural Architecture Search for Recommender Systems☆29Updated last year
- Prototype routines for GPU quantization written using PyTorch.☆19Updated last week
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 11 months ago
- ☆11Updated 6 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆89Updated this week
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"☆59Updated 4 months ago
- MACTA: A Multi-agent Reinforcement Learning Approach for Cache Timing Attacks and Detection☆46Updated last year
- Triton kernels for Flux☆20Updated last month
- ☆87Updated this week
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆38Updated 11 months ago
- ☆46Updated last year
- Repository for CPU Kernel Generation for LLM Inference☆25Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- sigma-MoE layer☆18Updated last year
- ☆42Updated last year
- FlexAttention w/ FlashAttention3 Support☆26Updated 4 months ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆64Updated 2 months ago
- DPO, but faster 🚀☆34Updated 2 months ago
- Patch convolution to avoid large GPU memory usage of Conv2D☆85Updated 3 weeks ago
- Exploration into the Firefly algorithm in Pytorch☆35Updated last week
- Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)☆18Updated last year
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆45Updated last year
- ☆30Updated 2 months ago
- This is a simple torch implementation of the high performance Multi-Query Attention☆16Updated last year