VoltaML / volta-treesLinks
☆22Updated 2 years ago
Alternatives and similar repositories for volta-trees
Users that are interested in volta-trees are comparing it to the libraries listed below
Sorting:
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- Running Stable Diffusion with Metaflow☆33Updated 3 months ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 2 years ago
- ☆32Updated 2 years ago
- ⚡VoltaML is a lightweight library to convert and run your ML/DL deep learning models in high performance inference runtimes like TensorRT…☆1,186Updated 2 years ago
- Implementation of denoising diffusion models with schedules, improved sampling, and other extensions using Keras.☆120Updated last year
- Benchmarking PyTorch 2.0 different models☆20Updated 2 years ago
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆92Updated 2 years ago
- Notes and plans for fastdiffusion course☆209Updated 2 years ago
- Train fastai models faster (and other useful tools)☆71Updated 3 months ago
- Collection of Pytorch lightning tutorial form as rich scripts automatically transformed to ipython notebooks.☆319Updated last month
- PyTorch interface for the IPU☆181Updated last year
- Playing around with stable diffusion. Generated images are reproducible because I save the metadata and latent information. You can gener…☆205Updated 3 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated 2 years ago
- Automatic gradient descent☆210Updated 2 years ago
- Prototype routines for GPU quantization written using PyTorch.☆21Updated last month
- Implementation of Flash Attention in Jax☆216Updated last year
- FasterAI: Prune and Distill your models with FastAI and PyTorch☆249Updated 3 months ago
- Stride visualizations☆38Updated 7 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated 2 months ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- Accelerate PyTorch models with ONNX Runtime☆364Updated 6 months ago
- JAX implementation ViT-VQGAN☆59Updated 3 years ago
- The fastai book, 2nd edition (in progress)☆56Updated last year
- ☆17Updated 3 years ago
- ☆57Updated 3 years ago
- Fourth place solution to the "OpenVaccine: COVID-19 mRNA Vaccine Degradation Prediction" organized by Stanford University and Kaggle☆20Updated 4 years ago
- Text to Image Diffusion Models in Keras☆77Updated last year
- ☆21Updated 6 months ago