optimum
https://github.com/huggingface/optimum
Python
🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported2 Subscribers
Add a CodeTriage badge to optimum
Help out
- Issues
- Add BetterTransformer support for ESM Huggingface model for protein folding
- ONNX support for decision transformers
- [ONNX] Use the `dynamo=True` option from PyTorch 2.5
- Update ort CIs (slow, gpu, train)
- Add ONNX export support for DETA model
- get_wikitext2 has bug
- Adding Support for DETA Model
- [WIP] Refactor to Introduce Backend Abstraction
- Optional `subfolder` if model repository contains one ONNX model behind a subfolder
- [Feature request] Add kwargs or additional options for torch.onnx.export
- Docs
- Python not yet supported