Despite some limitations, you will probably find 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"