Bumblebee provides pre-trained Neural Network models on top of Axon. It includes integration with Models, allowing anyone to download and perform Machine Learning tasks with few lines of code. The best way to get started with Bumblebee is with Livebook. Our announcement video shows how to use Livebook's Smart Cells to perform different Neural Network tasks with a few clicks. You can then tweak the code and deploy it. First, add Bumblebee and EXLA as dependencies in your mix.exs. EXLA is an optional dependency but an important one as it allows you to compile models just-in-time and run them on CPU/GPU.
Features
- Pre-trained Neural Network models in Axon
- Documentation available
- Examples available
- We also provide single-file examples of running Neural Networks inside your Phoenix
- HuggingFace Hub
- Tokenizer support
Categories
Neural Network LibrariesLicense
Apache License V2.0Follow Bumblebee
You Might Also Like
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of Bumblebee!