Guidebook and reference on PyTorch training optimizations
☆27Sep 24, 2022Updated 3 years ago
Alternatives and similar repositories for pytorch-training-performance-guide
Users that are interested in pytorch-training-performance-guide are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- II-Thought-RL is our initial attempt at developing a large-scale, multi-domain Reinforcement Learning (RL) dataset☆31Apr 8, 2025Updated 11 months ago
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆18Feb 9, 2026Updated last month
- PyTorch implementation for PaLM: A Hybrid Parser and Language Model.☆10Jan 7, 2020Updated 6 years ago
- ☆15Jun 30, 2021Updated 4 years ago
- Jupyter for Data Science, published by Packt☆20Jan 30, 2023Updated 3 years ago
- Hacks for PyTorch☆19Apr 18, 2023Updated 2 years ago
- ☆10Apr 2, 2024Updated last year
- The code for Generative Locally Linear Embedding (GLLE).☆13Nov 4, 2021Updated 4 years ago
- ☆16Feb 24, 2026Updated last month
- Official repository for "Stylized Adversarial Training" (TPAMI 2022)☆11Dec 30, 2022Updated 3 years ago
- This is an official implementation of our CVPR 2020 paper "Non-Local Neural Networks With Grouped Bilinear Attentional Transforms".☆12Jan 30, 2021Updated 5 years ago
- High Performance FP8 GEMM Kernels for SM89 and later GPUs.☆20Jan 24, 2025Updated last year
- ☆11Aug 1, 2024Updated last year
- [SIGIR 2025] The official repo for "Scaling Sparse and Dense Retrieval in Decoder-Only LLMs"☆20Mar 31, 2025Updated 11 months ago
- A comprehensive overview of Data Distillation and Condensation (DDC). DDC is a data-centric task where a representative (i.e., small but …☆13Dec 1, 2022Updated 3 years ago
- Personal solutions to the Triton Puzzles☆20Jul 18, 2024Updated last year
- A package for plotting and manipulating 1D spectra☆12Jul 30, 2024Updated last year
- Lowering PyTorch's Memory Consumption for Selective Differentiation☆12Aug 29, 2024Updated last year
- ☆10Dec 1, 2017Updated 8 years ago
- Learn from amazing Kagglers on Kaggle☆12Feb 26, 2023Updated 3 years ago
- Python library for hyperspectral analysis focused on spectroscopic approach.☆18Sep 9, 2024Updated last year
- Curse-of-memory phenomenon of RNNs in sequence modelling☆19May 8, 2025Updated 10 months ago
- ☆12Oct 24, 2024Updated last year
- Collective and Neighbor Collective Optimizations and Extensions☆13Mar 2, 2026Updated 3 weeks ago
- ☆19Mar 3, 2025Updated last year
- [WSDM 2024 Best Paper Honorable Mention] Debiasing Sequential Recommenders through Distributionally Robust Optimization over System Expos…☆15Jun 20, 2024Updated last year
- [UbiComp 2020] Intermittent Learning: On-Device Machine Learning on Intermittently Powered System☆10May 28, 2020Updated 5 years ago
- Code for paper Extended Multiplicative Signal Augmentation: An alternative to pre-processing of infrared spectra☆14Jan 15, 2021Updated 5 years ago
- Pure Triton kernels for Qwen3.5-27B inference on NVIDIA B200☆81Feb 28, 2026Updated 3 weeks ago
- ☆13Jul 29, 2024Updated last year
- Implementation of Spectral Leakage and Rethinking the Kernel Size in CNNs in Pytorch☆14Feb 3, 2021Updated 5 years ago
- A high-performance attention mechanism that computes softmax normalization in a single streaming pass using running accumulators (online …☆29Oct 11, 2025Updated 5 months ago
- TorchEsegeta: Interpretability and Explainability pipeline for PyTorch☆20Feb 19, 2024Updated 2 years ago
- Replication files for my book "Modern Dimension Reduction"☆17Jan 2, 2022Updated 4 years ago
- Learning adapter weights from task descriptions☆19Nov 12, 2023Updated 2 years ago
- A comprehensive collection of Python decorators and utility functions to enhance your development workflow.☆27Jan 21, 2025Updated last year
- [ICLR 2021: Spotlight] Source code for the paper "A Panda? No, It's a Sloth: Slowdown Attacks on Adaptive Multi-Exit Neural Network Infer…☆15Feb 16, 2022Updated 4 years ago
- [Applied Intelligence 2022] Python code for ACP☆12Sep 5, 2023Updated 2 years ago
- A python package for robust statistics☆22Sep 9, 2025Updated 6 months ago