Allow pytorch backend when available in PyMC

pymc-devs / nutpie

Python wrapper for nuts-rs

MIT License

107 stars 10 forks source link

Allow pytorch backend when available in PyMC #128

Open ricardoV94 opened 3 months ago

ricardoV94 commented 3 months ago

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?

aseyboldt commented 3 months ago

Coming up soon :-)

aseyboldt commented 1 month ago

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