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:
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated 6 months ago
- Proposal for the next generation of course-oriented IR.☆10Dec 24, 2021Updated 4 years ago
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 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
- Desktop version of ChatGPT, support manually set cookie☆19Dec 9, 2022Updated 3 years ago
- 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
- Manages vllm-nccl dependency☆17Jun 3, 2024Updated last year
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Jun 8, 2023Updated 2 years ago
- Primo: Practical Learning-Augmented Systems with Interpretable Models☆19Dec 26, 2023Updated 2 years ago
- ☆23Jan 7, 2022Updated 4 years ago
- Python3 auto-active verification library (migrated to an Intel project)☆25Apr 7, 2022Updated 3 years ago
- Surrogate-based Hyperparameter Tuning System☆28Jun 29, 2023Updated 2 years ago
- A memory efficient DLRM training solution using ColossalAI☆107Nov 22, 2022Updated 3 years ago
- Artifact for "Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Serving" [SOSP '24]☆25Nov 21, 2024Updated last year
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆33Nov 29, 2024Updated last year
- 兆京大学班车预约 for Humans™☆30Nov 14, 2025Updated 3 months ago
- Research and development for optimizing transformers☆131Feb 16, 2021Updated 5 years ago
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Nov 22, 2022Updated 3 years ago
- Code accompanying the NeurIPS 2019 paper AutoAssist: A Framework to Accelerate Training of Deep Neural Networks.☆14Oct 3, 2022Updated 3 years ago
- netbeacon - monitoring your network capture, NIDS or network analysis process☆19Oct 26, 2013Updated 12 years ago
- [ICDCS 2023] Evaluation and Optimization of Gradient Compression for Distributed Deep Learning☆10Apr 28, 2023Updated 2 years ago
- ☆13Nov 15, 2017Updated 8 years ago
- ☆41Jun 18, 2021Updated 4 years ago
- Performance benchmarking with ColossalAI☆38Jul 6, 2022Updated 3 years ago
- A tool for cross-checking Verilog compilers☆14Apr 16, 2025Updated 10 months ago
- ☆20Jul 29, 2024Updated last year
- ☆12May 18, 2024Updated last year
- ☆15Jul 18, 2023Updated 2 years ago
- A Coq framework to support structural design and proof of hardware cache-coherence protocols☆14May 7, 2022Updated 3 years ago
- Chaitin-Briggs register-allocation algorithm (LLVM back-end)☆12Jan 6, 2016Updated 10 years ago
- ☆11Mar 13, 2023Updated 2 years ago
- ☆11Sep 22, 2017Updated 8 years ago
- For our ISSTA'23 paper ACETest: Automated Constraint Extraction for Testing Deep Learning Operators☆13Mar 30, 2024Updated last year
- CK-NNTest: collaboratively validating, benchmarking and optimizing neural net operators across platforms, frameworks and datasets☆13Jul 10, 2021Updated 4 years ago
- How to plot for papers, slides, demos, etc.☆10Apr 7, 2022Updated 3 years ago
- FPGA-based HyperLogLog Accelerator☆12Jul 13, 2020Updated 5 years ago
- Segmented Code Adjustment Quantization (SAQ)☆17Sep 22, 2025Updated 5 months ago
- Distributed, Replicated, Protocol-generic Key-value Store in Async Rust For SMR Protocols Research☆17Feb 10, 2026Updated 2 weeks ago
- A brief tutorial for eBPF: Verifier, observability, networking, and security.☆12Sep 19, 2024Updated last year