PyTorch implementation of LAMB for ImageNet/ResNet-50 training
☆13May 13, 2021Updated 4 years ago
Alternatives and similar repositories for pytorch-lamb
Users that are interested in pytorch-lamb are comparing it to the libraries listed below
Sorting:
- Accuracy 77%. Large batch deep learning optimizer LARS for ImageNet with PyTorch and ResNet, using Horovod for distribution. Optional acc…☆38Jun 1, 2021Updated 4 years ago
- ☆24Oct 14, 2022Updated 3 years ago
- Concise Reasoning via Reinforcement Learning☆13Apr 16, 2025Updated 10 months ago
- [KDD 2023] code for "Test accuracy vs. generalization gap: model selection in NLP without accessing training or testing data" https://arx…☆12Oct 17, 2022Updated 3 years ago
- The official repository for ICASSP 2020 Transformer VAE paper☆12Apr 25, 2023Updated 2 years ago
- Helpful utility APIs for PyTorch projects.☆10Jul 25, 2024Updated last year
- PyTorch Implementation of Improving Training Stability for Multitask Ranking Models in Recommender Systems☆11Oct 29, 2023Updated 2 years ago
- Neural network approximators of linear algebra operations on GPU with PyTorch☆17May 30, 2022Updated 3 years ago
- Host CIFAR-10.2 Data Set☆13Sep 22, 2021Updated 4 years ago
- A dataset with classified film shots☆11Aug 8, 2022Updated 3 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆125Nov 27, 2024Updated last year
- ☆14Nov 28, 2022Updated 3 years ago
- [ICML2024] Learning Divergence Fields for Shift-Robust Graph Representations☆11Aug 15, 2024Updated last year
- PointHR: Exploring High-Resolution Architectures for 3D Point Cloud Segmentation☆13Oct 13, 2023Updated 2 years ago
- Official PyTorch implementation of "Rethinking Value Function Learning for Generalization in Reinforcement Learning" (NeurIPS 2022)☆14Feb 20, 2023Updated 3 years ago
- Shows how to create basic image adversaries, and train adversarially robust image classifiers (to some extent).☆13Oct 14, 2020Updated 5 years ago
- The official implementation of "Helen: Optimizing CTR Prediction Models with Frequency-wise Hessian Eigenvalue Regularization"☆16Mar 14, 2024Updated last year
- PolyGen and Sceneformer in PyTorch☆15Jul 6, 2022Updated 3 years ago
- ☆14Sep 29, 2021Updated 4 years ago
- PyTorch implementation of StableMask (ICML'24)☆15Jun 27, 2024Updated last year
- A PyTorch Implementation of Neural Turing Machine☆13Jul 24, 2020Updated 5 years ago
- ☆18Nov 10, 2024Updated last year
- A set of GPU optimized space-filling-curve algorithms (e.g., Hilbert, Z-order) on PyTorch and OpenAI Triton.☆18Oct 31, 2024Updated last year
- Open AI Gym version of Berkeley AI Pacman with images as states☆13May 4, 2018Updated 7 years ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Jun 5, 2018Updated 7 years ago
- MLPruning, PyTorch, NLP, BERT, Structured Pruning☆20Jun 29, 2021Updated 4 years ago
- Official implementations for "Long Range Pooling for 3D Large-Scale Scene Understanding" (CVPR 2023)☆16Apr 24, 2023Updated 2 years ago
- ☆18Dec 6, 2021Updated 4 years ago
- Code for Neural Execution Engines: Learning to Execute Subroutines☆18Jan 11, 2021Updated 5 years ago
- Please go to https://github.com/facebookresearch/stable_signature☆15Jul 26, 2023Updated 2 years ago
- Tiny-Megatron, a minimalistic re-implementation of the Megatron library☆23Sep 1, 2025Updated 6 months ago
- ☆17Aug 3, 2022Updated 3 years ago
- Instructions and templates for SC authors☆17Aug 22, 2021Updated 4 years ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19May 10, 2024Updated last year
- Breaking the Curse of Space Explosion: Towards Efficient NAS with Curriculum Search☆16Jul 25, 2024Updated last year
- Research project - real-time multi-agent pursuit a moving target☆17Mar 13, 2021Updated 4 years ago
- This repository contains code for the MicroAdam paper.☆21Dec 14, 2024Updated last year
- ☆17Dec 11, 2022Updated 3 years ago
- ☆19Jul 6, 2023Updated 2 years ago