davidrpugh / horovod-gpu-data-science-project
Template repository for a Python 3-based data science project that uses Horovod.
☆43Updated 3 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
- Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"☆31Updated 4 years ago
- ☆37Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆37Updated 4 years ago
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆104Updated 3 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Functional deep learning☆108Updated 2 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- Convert scikit-learn models to PyTorch modules☆162Updated 11 months ago
- Differentiable Sorting Networks☆114Updated last year
- ☆73Updated 2 years ago
- GPT, but made only out of MLPs☆88Updated 3 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆67Updated 2 years ago
- MLBench Framework Core Python Library☆17Updated 2 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆78Updated 4 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆49Updated last year
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- Cyclemoid implementation for PyTorch☆89Updated 3 years ago
- ☆19Updated 3 years ago
- ☆55Updated last year
- ☆54Updated 4 years ago
- ☆25Updated 4 years ago
- Torch Distributed Experimental☆115Updated 8 months ago
- Integrate pytorch☆46Updated 4 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- PyTorch implementation of FNet: Mixing Tokens with Fourier transforms☆26Updated 3 years ago
- python library, scripts and notebooks that are usfull from time to time☆44Updated 3 years ago
- ☆14Updated 2 years ago