4 ms·
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
by eduardoleao052 3y ago
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!
- deleted 3y ago[deleted]
- krohling 3y agoThis is awesome! Congrats. Clearly a lot of work and a very cool library.
- eduardoleao052 3y agoThank you very much for the feedback! If you have any question about it, let me know.
- sroussey 3y agoYou 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 https://github.com/ashvardanian/SimSIMD I haven't looked at ffi over node yet.
- eduardoleao052 3y agoThanks for the tip! I will look into it. Speed in the tensor calculations is definitely one of the largest challenges in this project.