louis-she / pytorch-tao
tao for machine learning
☆12Updated last year
Alternatives and similar repositories for pytorch-tao:
Users that are interested in pytorch-tao are comparing it to the libraries listed below
- ☆15Updated 3 years ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- Benchmarking PyTorch 2.0 different models☆21Updated 2 years ago
- Hacks for PyTorch☆19Updated 2 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆35Updated 3 years ago
- Experiments for the blog post "No, We Don't Have to Choose Batch Sizes As Powers Of 2"☆20Updated 2 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.☆13Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 6 months ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 3 years ago
- A dashboard for exploring timm learning rate schedulers☆19Updated 5 months ago
- Distributed DataLoader For Pytorch Based On Ray☆24Updated 3 years ago
- ☆38Updated 2 years ago
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆22Updated last month
- A collection of reproducible inference engine benchmarks☆24Updated this week
- WIP. Veloce is a low-code Ray-based parallelization library that makes machine learning computation novel, efficient, and heterogeneous.☆18Updated 2 years ago
- Experiments with Message Passing GNNs in C++ and PyTorch.☆25Updated 9 months ago
- Companion code for a tutorial on using Hydra.☆29Updated 3 years ago
- This repository contains notebooks showing how to perform mixed precision training in tf.keras 2.0☆12Updated 5 years ago
- Personal collection of references for high performance mixed precision training.☆41Updated 5 years ago
- Optimization and Generalization Analysis of Transduction through Gradient Boosting and Application to Multi-scale Graph Neural Networks☆13Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆63Updated 3 weeks ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- PyTorch implementation of GLOM☆22Updated 3 years ago
- Benchmark for machine learning model online serving (LLM, embedding, Stable-Diffusion, Whisper)☆28Updated last year
- A "gym" style toolkit for building lightweight NAS systems.☆13Updated 2 years ago
- Some microbenchmarks and design docs before commencement☆12Updated 4 years ago
- Describe the format of image/text datasets☆11Updated 3 years ago