PyTorch?

[watts4speed]

Is there a PyTorch implementation available or one being planned? My work is in PyTorch and it's not practical to convert over to JAX.

Thanks

[bhoov]

I tried a few years ago when pytorch's functional grad interface was nascent and resulted in error-prone usage of these networks.

I hear pytorch's interface has developed a lot since then and it shouldn't take long to make one from scratch. Stay tuned

[bhoov]

Barebones repo completed, please see hamux-torch and its accompanying example example_mnist.ipynb

Almost identical API. Pytorch functional has come a long way