Deep Learning in JavaScript, with GPU Support

Posted by eduardoleao052 on Saturday July 20, 2024 @06:50PM
from the github.com dept.

https://github.com/eduardoleao052/js-pytorch

( 3 comments | 4 points | on HN )

Re:Deep Learning in JavaScript, with GPU Support by eduardoleao052 on Saturday July 20, 2024 @07:01PM (#41020831)

Three months ago I made a post here with an early version of this project, and received some great feedback! One of the most asked features was GPU support, so I added it alongside many other improvements. Hope you like it!

[ Parent | on HN ]

Re:Deep Learning in JavaScript, with GPU Support by IAmBurger on Saturday July 20, 2024 @06:54PM (#41020777)

Really cool stuff! Does it have save/load functions?

[ Parent | on HN ]

Re:Deep Learning in JavaScript, with GPU Support by eduardoleao052 on Saturday July 20, 2024 @06:58PM (#41020801)

Thanks for the feedback, glad you liked it! It does, and they work similarly to their PyTorch counterparts. The explanation is on the “Running it Yourself” part of the README.

[ Parent | on HN ]