PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.
☆10Feb 10, 2022Updated 4 years ago
Alternatives and similar repositories for PSTensor
Users that are interested in PSTensor are comparing it to the libraries listed below
Sorting:
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated 7 months ago
- Desktop version of ChatGPT, support manually set cookie☆19Dec 9, 2022Updated 3 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆133Jul 6, 2023Updated 2 years ago
- Proposal for the next generation of course-oriented IR.☆10Dec 24, 2021Updated 4 years ago
- Manages vllm-nccl dependency☆17Jun 3, 2024Updated last year
- WIP. Veloce is a low-code Ray-based parallelization library that makes machine learning computation novel, efficient, and heterogeneous.☆17Aug 4, 2022Updated 3 years ago
- Primo: Practical Learning-Augmented Systems with Interpretable Models☆19Dec 26, 2023Updated 2 years ago
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Jun 8, 2023Updated 2 years ago
- ☆18Jan 27, 2025Updated last year
- ☆23Jan 7, 2022Updated 4 years ago
- A memory efficient DLRM training solution using ColossalAI☆107Nov 22, 2022Updated 3 years ago
- Surrogate-based Hyperparameter Tuning System☆29Jun 29, 2023Updated 2 years ago
- ☆21Jul 10, 2025Updated 8 months ago
- FGNN's artifact evaluation (EuroSys 2022)☆18Apr 25, 2022Updated 3 years ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆33Nov 29, 2024Updated last year
- Fast Synchronization-Free Algorithms for Parallel Sparse Triangular Solves with Multiple Right-Hand Sides (SpTRSM)☆14Feb 14, 2020Updated 6 years ago
- GPU-scheduler-for-deep-learning☆209Nov 5, 2020Updated 5 years ago
- GPT Demo with hybrid distributed training☆10Dec 1, 2022Updated 3 years ago
- 兆京大学班车预约 for Humans™☆30Nov 14, 2025Updated 4 months ago
- ☆25Nov 20, 2025Updated 4 months ago
- ☆26Aug 19, 2022Updated 3 years ago
- Python3 auto-active verification library (migrated to an Intel project)☆25Apr 7, 2022Updated 3 years ago
- Hanzi to Pinyin engine in Swift 拼音输入法引擎☆14Mar 29, 2024Updated last year
- A Translation Task using TurboTransformers☆10Dec 17, 2020Updated 5 years ago
- ☆41Jun 18, 2021Updated 4 years ago
- The Firmament cluster scheduling platform☆19Mar 15, 2019Updated 7 years ago
- ☆54Dec 10, 2025Updated 3 months ago
- Drop-in library for tracking the memory allocations of CUDA applications☆14Nov 17, 2017Updated 8 years ago
- Performance benchmarking with ColossalAI☆39Jul 6, 2022Updated 3 years ago
- Research and development for optimizing transformers☆131Feb 16, 2021Updated 5 years ago
- Artifact for OSDI'23: MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Mult…☆41Mar 17, 2024Updated 2 years ago
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Nov 22, 2022Updated 3 years ago
- ☆11May 15, 2025Updated 10 months ago
- ☆20Nov 7, 2019Updated 6 years ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Dec 20, 2019Updated 6 years ago
- Tiresias is a GPU cluster manager for distributed deep learning training.☆166May 7, 2020Updated 5 years ago
- Code for EMNLP 2020 paper: Analogous Process Structure Induction for Sub-event Sequence Prediction☆11Oct 19, 2020Updated 5 years ago
- Spatio-temporal pattern contruct and model fusion☆11Jun 10, 2019Updated 6 years ago