The-AI-Summer / pytorch-ddpLinks
code for the ddp tutorial
☆32Updated 3 years ago
Alternatives and similar repositories for pytorch-ddp
Users that are interested in pytorch-ddp are comparing it to the libraries listed below
Sorting:
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆39Updated 3 years ago
- PyTorch implementation of Soft MoE by Google Brain in "From Sparse to Soft Mixtures of Experts" (https://arxiv.org/pdf/2308.00951.pdf)☆82Updated 2 years ago
- ☆133Updated 2 years ago
- Adversarial examples to the new ConvNeXt architecture☆20Updated 4 years ago
- PyTorch, PyTorch Lightning framework for trying knowledge distillation in image classification problems☆32Updated last year
- ☆37Updated 2 years ago
- ImageNet-12k subset of ImageNet-21k (fall11)☆21Updated 2 years ago
- Code for the paper "Query-Key Normalization for Transformers"☆50Updated 4 years ago
- Implementation of Infini-Transformer in Pytorch☆112Updated last year
- This repository holds code and other relevant files for the NeurIPS 2022 tutorial: Foundational Robustness of Foundation Models.☆72Updated 3 years ago
- several types of attention modules written in PyTorch for learning purposes☆52Updated 3 weeks ago
- PyTorch implementation of moe, which stands for mixture of experts☆52Updated 4 years ago
- A simple implementation of a deep linear Pytorch module☆21Updated 5 years ago
- Implementation of Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer in PyTorch.☆53Updated 2 years ago
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆56Updated 2 years ago
- Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)☆63Updated 3 years ago
- ☆29Updated 3 years ago
- CIFAR10 ResNets implemented in JAX+Flax☆12Updated 3 years ago
- ☆34Updated last year
- Implementation of Cross Transformer for spatially-aware few-shot transfer, in Pytorch☆54Updated 4 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 3 years ago
- Code for the PAPA paper☆27Updated 3 years ago
- Triton Implementation of HyperAttention Algorithm☆48Updated 2 years ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆16Updated 4 years ago
- ☆64Updated last year
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆122Updated last year
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆131Updated last year
- [COLM 2024] Early Weight Averaging meets High Learning Rates for LLM Pre-training☆18Updated last year
- ☆41Updated last year
- A regression-alike loss to improve numerical reasoning in language models - ICML 2025☆27Updated 5 months ago