aieater / rocm_pytorch_informations
The official page of ROCm/PyTorch will contain information that is always confusing. On this page we will endeavor to describe accurate information based on the knowledge gained by GPUEater infrastructure development.
☆87Updated 4 years ago
Alternatives and similar repositories for rocm_pytorch_informations:
Users that are interested in rocm_pytorch_informations are comparing it to the libraries listed below
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆221Updated this week
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆127Updated last year
- Trains Transformer model variants. Data isn't shuffled between batches.☆139Updated 2 years ago
- Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …☆106Updated last month
- Haste: a fast, simple, and open RNN library☆328Updated last year
- Synthesize image datasets of documents in natural scenes with Python+Blender3D☆57Updated 2 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆83Updated 3 years ago
- ☆52Updated 4 years ago
- Code for scaling Transformers☆26Updated 4 years ago
- Using the famous cnn model in Pytorch, we run benchmarks on various gpu.☆228Updated 7 months ago
- Mish Activation Function for PyTorch☆148Updated 4 years ago
- ☆148Updated 2 years ago
- Large Model Support in PyTorch☆133Updated 2 years ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆205Updated last year
- HetSeq: Distributed GPU Training on Heterogeneous Infrastructure☆106Updated last year
- Babysit your preemptible TPUs☆85Updated 2 years ago
- nGraph™ Backend for ONNX☆42Updated 2 years ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 3 years ago
- GPU fan control for headless Linux☆338Updated last year
- Python way to Read/Write TFRecords☆64Updated 6 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆371Updated 4 years ago
- A simple and working implementation of Electra, the fastest way to pretrain language models from scratch, in Pytorch☆221Updated last year
- My implementation of DeepMind's Perceiver☆61Updated 3 years ago
- Torch Distributed Experimental☆115Updated 6 months ago
- A queue service for quickly developing scripts that use all your GPUs efficiently☆83Updated 2 years ago
- Parameterized fit and prediction harnesses for pytorch☆39Updated 4 years ago
- Fully featured implementation of Routing Transformer☆288Updated 3 years ago
- ☆109Updated 3 years ago
- NVIDIA GPU tools - monitoring on CLI & web app with multiple agents☆87Updated 9 months ago
- Python package containing all custom layers used in Neural Networks (Compatible with PyTorch, TensorFlow and MegEngine)☆137Updated last year