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 5 years ago
- ☆17Jun 19, 2023Updated 3 years ago
- ☆10Nov 6, 2024Updated last year
- RAPIDS Deployment Documentation☆15Updated this week
- Streamline data pipelines for AI. Process datasets across 1000s of machines, and optimize data for blazing fast model training.☆16Sep 18, 2024Updated 2 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.
- ☆130Oct 25, 2023Updated 2 years ago
- ☆15Feb 13, 2018Updated 8 years ago
- Concept bottleneck models for multiview data with incomplete concept sets☆17Nov 24, 2023Updated 2 years ago
- Comparing Deep Learning Inference of Pytorch models running on CPU, CUDA and TensorRT☆17Feb 20, 2022Updated 4 years ago
- Distilling key points, reorganizing, and modestly augmenting the points from books and lectures.☆12Oct 1, 2026Updated last week
- AI-powered browser extension to chat with any webpage☆10Aug 12, 2025Updated last year
- Supervised instruction finetuning for LLM with HF trainer and Deepspeed☆37Jul 6, 2023Updated 3 years ago
- ☆27Mar 15, 2023Updated 3 years ago
- Python intefrace for evaluation on chatgpt models☆19Feb 13, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- a version of baby agi using dspy and typed predictors☆15Mar 9, 2024Updated 2 years ago
- ☆14Apr 10, 2023Updated 3 years ago
- ☆263Nov 24, 2025Updated 10 months ago
- This is a repository for the paper: "Sequence anticipation and spike-timing-dependent plasticity emerge from a predictive learning rule" …☆12Jun 2, 2024Updated 2 years ago
- ☆54Jul 18, 2024Updated 2 years ago
- Source code for COLING 2022 paper "Automatic Label Sequence Generation for Prompting Sequence-to-sequence Models"☆23Sep 21, 2022Updated 4 years ago
- ☆19Nov 24, 2025Updated 10 months ago
- QLoRA for Masked Language Modeling☆23Sep 11, 2023Updated 3 years ago
- ☆28Apr 26, 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.
- Code for paper DMCVR: Morphology-Guided Diffusion Model for 3D Cardiac Volume Reconstruction☆14Jan 12, 2024Updated 2 years ago
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆35May 1, 2025Updated last year
- Handy list of network visualisation libraries for R☆12Nov 11, 2019Updated 6 years ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Jul 28, 2023Updated 3 years ago
- Testing paligemma2 finetuning on reasoning dataset☆18Dec 28, 2024Updated last year
- A python (PyTorch) implementation of Polar Transform Network (PTN) method for prostate ultrasound segmentation☆14Nov 5, 2024Updated last year
- ☆11Aug 22, 2023Updated 3 years ago
- 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
- Chrome Extension for YouTube. Acts as an assistant for the YouTube video you are watching☆22Apr 26, 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.
- A tiny server to run local inference on MLX model in the style of OpenAI☆13Jan 31, 2024Updated 2 years ago
- Implementations of 2D Image Convolution algorithm with CUDA (using global memory, shared memory and constant memory)☆17Jan 21, 2018Updated 8 years ago
- All my experiments with the various transformers and various transformer frameworks available☆14Apr 30, 2021Updated 5 years ago
- ☆19Sep 26, 2023Updated 3 years ago
- Class for producing smooth 17 and 18 segment left ventricle plot, recommended by American Heart Association☆19Nov 2, 2023Updated 2 years ago
- A notebook testing CPU speed vs GPU speed with Pytorch and CUDA☆18Dec 25, 2021Updated 4 years ago
- ☆11Nov 5, 2021Updated 4 years ago