nutpie
nutpie copied to clipboard
Allow pytorch backend when available in PyMC
The barebones PyTorch backend should be in in the next PyTensor release and subsequent PyMC.
I think the best way to test it out and gather info on what Ops are most needed is to give access from nutpie, as it will:
- Feel like experimental on experimental
- Be tested by more code-savy people
What do you think?
Coming up soon :-)
Current work is here: https://github.com/pymc-devs/nutpie/pull/144