Functional local implementations of main model parallelism approaches
☆95Feb 21, 2023Updated 3 years ago
Alternatives and similar repositories for large-model-parallelism
Users that are interested in large-model-parallelism are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Test pytorch code with minimal computational overhead☆26Jun 8, 2023Updated 3 years ago
- ☆10Oct 11, 2022Updated 4 years ago
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆62Jun 21, 2023Updated 3 years ago
- A collection of resources regarding the interplay between control, causality, latent structure and self-supervision.☆11Dec 25, 2020Updated 5 years ago
- [WIP] Transformer to embed Danbooru labelsets☆13Mar 31, 2024Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Converts Quora's new NLU dataset to SNLI txt/jsonl format, plus test/dev split, tokenization.☆14Jan 27, 2017Updated 9 years ago
- Official repository for the paper: "Trees with Attention for Set Prediction Tasks" (ICML21)☆10Jan 19, 2022Updated 4 years ago
- Learning to Avoid Errors in GANs by Input Space Manipulation (Code for paper)☆23Jul 7, 2017Updated 9 years ago
- Winner of Grand Prize in Sustainability & Meta Best Innovation Award. We mitigate overfishing through AI-augmented satellite imagery and …☆30Feb 20, 2023Updated 3 years ago
- Rust bindings for CTranslate2☆14Jun 21, 2023Updated 3 years ago
- A better data loading pipeline for training ML models☆10Nov 21, 2023Updated 2 years ago
- Performant, differentiable reinforcement learning☆126Aug 2, 2025Updated last year
- ☆10Jun 13, 2022Updated 4 years ago
- ☆67Dec 8, 2023Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- An annotated implementation of the Hyena Hierarchy paper☆34May 28, 2023Updated 3 years ago
- Code repository for the paper "Geometric Scattering for Graph Data Analysis"☆14Aug 26, 2019Updated 7 years ago
- ☆10Oct 21, 2021Updated 5 years ago
- ☆10Dec 30, 2023Updated 2 years ago
- Code for "Dynamic Discounted Counterfactual Regret Minimization", ICLR 2024 (Spotlight)☆19Apr 22, 2024Updated 2 years ago
- ☆40May 26, 2023Updated 3 years ago
- Create the perfect environment to work and relax.☆10Mar 10, 2024Updated 2 years ago
- PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization☆16Apr 27, 2020Updated 6 years ago
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆69Updated this week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Code related to different aspects of conformal learning☆18Jan 28, 2025Updated last year
- [ICML 2023] Official code for "DevFormer: A Symmetric Transformer for Context-Aware Device Placement"☆23Dec 7, 2024Updated last year
- What would you do with 1000 H100s...☆1,202Jan 10, 2024Updated 2 years ago
- ☆18Dec 19, 2019Updated 6 years ago
- generative neural network trained with physics knowledge☆14Mar 8, 2021Updated 5 years ago
- A simple synthetic dataset and baseline model for visual counting.☆12Jan 15, 2019Updated 7 years ago
- Code to reproduce experiments in "A Rainbow in Deep Network Black Boxes".☆12Oct 21, 2024Updated 2 years ago
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 6 years ago
- The enhanced version of ZEN, larger and more powerful.☆31Jul 22, 2022Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆28Oct 21, 2022Updated 4 years ago
- 🔥 [ICML 2026] KAGE-Bench: pure JAX 2D platformer RL benchmark for visual OOD generalization. Massively-parallel (vmap/JIT) RGB env with …☆13Jul 1, 2026Updated 4 months ago
- simple bibtex generator for any text with \cite{}☆30Jul 13, 2024Updated 2 years ago
- GFlowNet library specialized for graph & molecular data☆294Sep 15, 2026Updated last month
- Research Agora: Claude Code skills, benchmarks & tools for ML researchers — paper writing, citation verification, experiment tracking, La…☆17Updated this week
- ☆275Jan 24, 2025Updated last year
- Deep reinforcement learning framework for fast prototyping based on PyTorch☆14Mar 12, 2023Updated 3 years ago