[IAmBurger]

Really cool stuff! Does it have save/load functions?

[eduardoleao052]

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.