I don't know much about machine learning, but there is a new FB tensor library that you can use from JS (bun) and with nbb: https://twitter.com/borkdude/status/1584130632036548608