PrunaAI / pruna
Pruna is a model optimization framework built for developers, enabling you to deliver faster, more efficient models with minimal overhead.
☆587Updated this week
Alternatives and similar repositories for pruna:
Users that are interested in pruna are comparing it to the libraries listed below
- A curated list of materials on AI efficiency☆22Updated last month
- ☆150Updated 7 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆421Updated 3 months ago
- Transform datasets at scale. Optimize datasets for fast AI model training.☆436Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆270Updated 9 months ago
- (WIP) A small but powerful, homemade PyTorch from scratch.☆540Updated this week
- Repo centralising material for the ULiege course INFO9023 "Machine Learning Systems Design (MLOps)".☆130Updated last week
- Access to free kaggle compute power from your command line☆28Updated last year
- The boundary of neural network trainability is fractal☆196Updated last year
- Efficient optimizers☆185Updated this week
- For optimization algorithm research and development.☆503Updated this week
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆169Updated last week
- Best practices & guides on how to write distributed pytorch training code☆377Updated last month
- The AdEMAMix Optimizer: Better, Faster, Older.☆179Updated 6 months ago
- Fast State-of-the-Art Static Embeddings☆1,129Updated this week
- Exca - Execution and caching tool for python☆79Updated this week
- Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).☆220Updated this week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆84Updated last week
- UNet diffusion model in pure CUDA☆600Updated 9 months ago
- Repository for CARTE: Context-Aware Representation of Table Entries☆116Updated 2 weeks ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆114Updated last month
- Python package for conformal prediction☆486Updated 6 months ago
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆90Updated 2 months ago
- PyTorch video decoding☆465Updated this week
- Domain adaptation toolbox compatible with scikit-learn and pytorch☆111Updated last month
- Uncertainty quantification with PyTorch☆349Updated last week
- Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …☆1,314Updated this week
- Actually Robust Training - Tool Inspired by Andrej Karpathy "Recipe for training neural networks". It allows you to decompose your Deep…☆44Updated 11 months ago
- ☆208Updated 8 months ago
- Scalable and Performant Data Loading☆231Updated this week