Pruna AI

🌍 Join the Pruna AI community!

💜 Make AI models faster, cheaper, smaller, greener!

Pruna AI makes AI models faster, cheaper, smaller, and greener with the pruna package.

• It supports various models, including CV, NLP, audio, and graphs for predictive and generative AI.
• It supports various hardware, including GPU, CPU, Edge.
• It supports various compression algorithms, including quantization, pruning, distillation, caching, recovery, compilation, or factorization, among others.
• You can combine algorithms to find the optimal configuration and smash/compress your model.
• You can evaluate reliable quality and efficiency metrics of your base vs smashed/compressed models.

Set it up in minutes and compress your first models in a few lines of code!

⏩ How to get started?

You can smash your own models by installing pruna with pip:

pip install pruna

You can start with simple notebooks to experience efficiency gains with:

For more details on installation and free tutorials, check the Pruna AI documentation.

✨ Test our endpoints

Want to use our optimized models right away? Try them via our API for fast, easy access to Pruna-powered inference.

Try our models