nutpie icon indicating copy to clipboard operation
nutpie copied to clipboard

Allow pytorch backend when available in PyMC

Open ricardoV94 opened this issue 1 year ago • 2 comments

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:

  1. Feel like experimental on experimental
  2. Be tested by more code-savy people

What do you think?

ricardoV94 avatar Jun 21 '24 13:06 ricardoV94

Coming up soon :-)

aseyboldt avatar Jun 24 '24 19:06 aseyboldt

Current work is here: https://github.com/pymc-devs/nutpie/pull/144

aseyboldt avatar Aug 06 '24 15:08 aseyboldt