Comment by 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.