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
- ☆9Updated last year
- Benchmarking PyTorch 2.0 different models☆21Updated last year
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Distributed DataLoader For Pytorch Based On Ray☆24Updated 3 years ago
- Benchmark for machine learning model online serving (LLM, embedding, Stable-Diffusion, Whisper)☆27Updated last year
- Hacks for PyTorch☆18Updated last year
- A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.☆13Updated 2 years ago
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆18Updated 2 months ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆26Updated 3 months ago
- Tensorboard extension for Jupyterlab all in one☆88Updated 5 months ago
- Hyperparameter tuning via uncertainty modeling☆46Updated 8 months ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- Test pytorch code with minimal computational overhead☆25Updated last year
- Create a source of truth for ML model results and browse it on Papers with Code☆26Updated 3 years ago
- Experiments for the blog post "No, We Don't Have to Choose Batch Sizes As Powers Of 2"☆19Updated 2 years ago
- ☆9Updated 3 months ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆35Updated 3 years ago
- ☆18Updated last year
- A dashboard for exploring timm learning rate schedulers☆19Updated last month
- CIFAR10 ResNets implemented in JAX+Flax☆11Updated 2 years ago
- Companion code for a tutorial on using Hydra.☆29Updated 3 years ago
- Triton Server Component for lightning.ai☆14Updated last year
- ☆10Updated last year
- Code for a workshop hosted at the MLOps World Summit '22☆16Updated 2 years ago
- Implementation of a holodeck, written in Pytorch☆17Updated last year