This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog post.
☆94Jul 14, 2023Updated 3 years ago
Alternatives and similar repositories for pytorch-memory-optim
Users that are interested in pytorch-memory-optim are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Materials for "Transformers from the Ground Up" at PyData Jeddah on August 5, 2021☆20Aug 5, 2021Updated 4 years ago
- ☆17Jun 19, 2023Updated 3 years ago
- My study notes and hands-on projects for CUDA-based GPU programming☆13Dec 11, 2025Updated 7 months ago
- ☆10Nov 6, 2024Updated last year
- Modular, flexible, cross-platform workload profiling and characterization☆13Mar 1, 2021Updated 5 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆130Oct 25, 2023Updated 2 years ago
- ☆15Feb 13, 2018Updated 8 years ago
- ☆14Apr 6, 2025Updated last year
- Converting a deep neural network to integer-only inference in native C via uniform quantization and the fixed-point representation.☆26Jan 31, 2022Updated 4 years ago
- OpenGraph is an open-source graph processing benchmarking suite written in pure C/OpenMP. Integrated with Sniper simulator.☆11Apr 27, 2024Updated 2 years ago
- Scaling Sparse Fine-Tuning to Large Language Models☆19Jan 31, 2024Updated 2 years ago
- ☆27Mar 15, 2023Updated 3 years ago
- Supervised instruction finetuning for LLM with HF trainer and Deepspeed☆37Jul 6, 2023Updated 3 years ago
- Read custom dataset☆12Mar 31, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Python intefrace for evaluation on chatgpt models☆19Feb 13, 2024Updated 2 years ago
- Fast convolution algorithms with Python types☆10Nov 20, 2016Updated 9 years ago
- Singular Binarized Neural Network based on GPU Bit Operations (see our SC-19 paper)☆17Dec 9, 2020Updated 5 years ago
- ☆253Nov 24, 2025Updated 7 months ago
- ☆16Jun 12, 2023Updated 3 years ago
- Sharing the codebase and steps for artifact evaluation for ISCA 2023 paper☆16Feb 20, 2024Updated 2 years ago
- ☆54Jul 18, 2024Updated 2 years ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆33Jun 28, 2023Updated 3 years ago
- ☆19Nov 24, 2025Updated 7 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- QLoRA for Masked Language Modeling☆23Sep 11, 2023Updated 2 years ago
- ☆28Apr 26, 2023Updated 3 years ago
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆34May 1, 2025Updated last year
- ☆23Feb 16, 2022Updated 4 years ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆33Jul 28, 2023Updated 2 years ago
- FIWARE 401: IDM - Managing Users and Organizations☆10May 15, 2026Updated 2 months ago
- Testing paligemma2 finetuning on reasoning dataset☆18Dec 28, 2024Updated last year
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated 2 years ago
- Simple C++ reader for CIFAR-10 dataset☆17Apr 29, 2023Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- All my experiments with the various transformers and various transformer frameworks available☆14Apr 30, 2021Updated 5 years ago
- An implementation of several unsupervised object discovery models (Slot Attention, SLATE, GNM) in PyTorch with pre-trained models.☆15May 26, 2025Updated last year
- Code and annotation for the paper "Towards Accurate and Interpretable Surgical Skill Assessment: A Video-Based Method Incorporating Recog…☆12Jan 20, 2023Updated 3 years ago
- Chrome Extension for YouTube. Acts as an assistant for the YouTube video you are watching☆22Apr 26, 2023Updated 3 years ago
- Implementations of various parallel algorithms for matrix factorization (including DSGD++)☆16Jul 14, 2017Updated 9 years ago
- ☆11Nov 5, 2021Updated 4 years ago
- ☆30Jul 22, 2024Updated last year