trax icon indicating copy to clipboard operation
trax copied to clipboard
verified publisher icon google

PyTorch backend

Open briankosw opened this issue 5 years ago • 2 comments

I notice that so far there are JAX, TensorFlow, and NumPy backends. Are there any plans to have a PyTorch backend in the future?

briankosw avatar Jul 17 '20 14:07 briankosw

We are quite busy debugging the current backends for now, but if someone made a PR with a nicely working one, sure!

lukaszkaiser avatar Jul 29 '20 08:07 lukaszkaiser

@lukaszkaiser I want to do this can you assign it to me.

sparkingdark avatar Oct 14 '20 07:10 sparkingdark