Comment by TheAnig

Sound like a pretty neat introduction! This is exactly the kind of thing I needed, coming from tf/keras and looking to switch to pytorch for research