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.