2 ms·

Pytorch actually got an experimental Java API in version 1.4 (about two weeks ago), if you're interested.