VoltaML / volta-trees
☆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
- Lightning HPO & Training Studio App☆18Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 8 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆155Updated 3 months ago
- ⚡VoltaML is a lightweight library to convert and run your ML/DL deep learning models in high performance inference runtimes like TensorRT…☆1,192Updated 2 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆111Updated 4 months ago
- Benchmarking PyTorch 2.0 different models☆21Updated 2 years ago
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆54Updated this week
- Fast sparse deep learning on CPUs☆52Updated 2 years ago
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆133Updated 2 years ago
- ML/DL Math and Method notes☆59Updated last year
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆43Updated 5 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆182Updated 2 years ago
- Machine Learning Pipeline for Semantic Segmentation with TensorFlow Extended (TFX) and various GCP products☆95Updated last year
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆87Updated 3 years ago
- Productionize machine learning predictions, with ONNX or without☆65Updated last year
- Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops☆28Updated last year
- Torch Distributed Experimental☆115Updated 7 months ago
- Train fastai models faster (and other useful tools)☆65Updated 9 months ago
- Blazing fast training of 🤗 Transformers on Graphcore IPUs☆84Updated last year
- Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.☆37Updated last year
- GPU Environment Management for Visual Studio Code☆37Updated last year
- ☆43Updated 7 months ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- ML model training for edge devices☆162Updated last year
- Hyperparameter tuning via uncertainty modeling☆47Updated 10 months ago
- Quickest way to share everything about your research within a single app☆38Updated 5 months ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated last year
- ☆202Updated 3 years ago