3 ms·
Does Elixir ML ecosystem libs (Nx, Axon) provide some sort of interop with Python ecosystem? For example can I load or fine tune a model pre-trained in pytorch
by jatins 3y ago
Does Elixir ML ecosystem libs (Nx, Axon) provide some sort of interop with Python ecosystem?
For example can I load or fine tune a model pre-trained in pytorch/JAX in Axon? Or does everything need to be written from ground up in Elixir?
- ricketycricket 3y agoOrtex exists to run ONNX models: https://github.com/elixir-nx/ortex https://github.com/elixir-nx/ortex
- ac_alejos 3y agoNot directly but it supports the ONNX Runtime and has support for many of the models you might want through Bumblebee and Hugging Face https://github.com/elixir-nx/bumblebee https://github.com/elixir-nx/bumblebee
- thibaut_barrere 3y ago(sorry for the double post!)
- thibaut_barrere 3y agoDespite some limitations, you will probably find Bumblebee (https://github.com/elixir-nx/bumblebee https://github.com/elixir-nx/bumblebee) interesting. "Bumblebee provides pre-trained Neural Network models on top of Axon. It includes integration with HuggingFace Models, allowing anyone to download and perform Machine Learning tasks with few lines of code"