I've been waiting for a Pytorch solution in TS/JS for ages, this looks really cool. It does seem a little odd to start new JS projects these days and not use TS instead however. Was this a deliberate choice?