🌍 Join the Pruna AI community!

💜 Simply make AI models faster, cheaper, smaller, greener!

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

• It supports various models including CV, NLP, audio, graphs for predictive and generative AI.
• It supports various hardware including GPU, CPU, Edge.
• It supports various compression algortihms including quantization, pruning, distillation, caching, recovery, compilation that can be combined together.
• You can either play on your own with smash/compression configurations or let the smashing/compressing agent find the optimal configuration [Pro].
• You can evaluate reliable quality and efficiency metrics of your base vs smashed/compressed models. You can set it up in minutes and compress your first models in few lines of code!

⏩ How to get started?

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

pip install pruna

or directly from source.

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

For more details about installation, free tutorials and Pruna Pro tutorials, you can check the Pruna AI documentation.