On: Deep Learning in JavaScript

[eduardoleao052]

Learning Machine Learning, I’ve always been interested in PyTorch and its Automatic Backpropagation.

In this project, I tried to reimplement most of PyTorch in Javascript. It is implemented from scratch in a well-documented, unit tested, and interpretable way, so it can help other JavaScript learners get into Machine Learning!

Hope you enjoy!

[sroussey → eduardoleao052]

You might look into node api or ffi and add some native code to speed up some operations on the server.

I helped with this project to do that and get prebuilt binaries: https://github.com/ashvardanian/SimSIMD

I haven't looked at ffi over node yet.

[eduardoleao052 → sroussey]

Thanks for the tip! I will look into it. Speed in the tensor calculations is definitely one of the largest challenges in this project.

[krohling → eduardoleao052]

This is awesome! Congrats. Clearly a lot of work and a very cool library.

[eduardoleao052 → krohling]

Thank you very much for the feedback! If you have any question about it, let me know.