davidrpugh / horovod-gpu-data-science-projectLinks
Template repository for a Python 3-based data science project that uses Horovod.
☆43Updated 4 years ago
Alternatives and similar repositories for horovod-gpu-data-science-project
Users that are interested in horovod-gpu-data-science-project are comparing it to the libraries listed below
Sorting:
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 5 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Differentiable Sorting Networks☆125Updated 2 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated last year
- ☆54Updated 5 years ago
- A Machine Learning workflow for Slurm.☆151Updated 4 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆215Updated 2 years ago
- Companion code for a tutorial on using Hydra.☆30Updated 4 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆110Updated 4 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 4 years ago
- 🎲 Iterable dataset resampling in PyTorch☆92Updated 3 years ago
- Integrate pytorch☆46Updated 4 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- ☆133Updated 4 years ago
- ☆100Updated 4 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆129Updated last year
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated last year
- Code for Parameter Prediction for Unseen Deep Architectures (NeurIPS 2021)☆492Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Convert scikit-learn models to PyTorch modules☆168Updated last year
- ☆37Updated 3 years ago
- Codebase for Learning Invariances in Neural Networks☆96Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- Functional deep learning☆108Updated 3 years ago
- Cyanure: An Open-Source Toolbox for Empirical Risk Minimization☆64Updated 6 months ago
- python library, scripts and notebooks that are usfull from time to time☆44Updated 4 years ago
- Semantic Segmentation with Pytorch-Lightning☆63Updated 4 years ago
- Structured matrices for compressing neural networks☆67Updated 2 years ago