Help wanted: Continue PyTorch backend for PyTensor

Anyone interested in contributing a PyTorch backend to PyTensor? I started a PR here: Experimental: PyTorch backend by twiecki · Pull Request #457 · pymc-devs/pytensor · GitHub

It doesn’t work yet but it’s a good starting point for someone motivated. There are failing tests which point to the current problems.

How did I create this? I copied the JAX-backend and asked GPT to port the code to PyTorch.

3 Likes