Three months ago I made a post here with an early version of this project, and received some great feedback! One of the most asked features was GPU support, so I added it alongside many other improvements. Hope you like it!
Really cool stuff! Does it have save/load functions?
Thanks for the feedback, glad you liked it!
It does, and they work similarly to their PyTorch counterparts. The explanation is on the “Running it Yourself” part of the README.