Convert a pytorch model to native elixir


I have seen some references that there is work being done to help migrate pytorch models to elixir. I am wondering if there is any pointers folks could give me to some resources related to this? Recently Whisper was released and I am very keen to try to pull that into some of my projects and see how it performs, but I would really like to run in elixir land and not need to shell out to pytorch.

Was I misremembering that there was a project being worked to change pytorch models to be able to run in elixir?

I think the Axon library may be what you are after (on my phone so sorry for the brief answer)