Deep Learning in JavaScript

1. JS-PyTorch

   1 5 1,134 9.5 JavaScript

   A JavaScript library like PyTorch, with GPU acceleration.

   

2. CodeRabbit

   coderabbit.ai featured

   CodeRabbit: AI Code Reviews for Developers. Revolutionize your code reviews with AI. CodeRabbit offers PR summaries, code walkthroughs, 1-click suggestions, and AST-based analysis. Boost productivity and code quality across all major languages with each PR.

   

3. lab-grad

   2 1 20 - TypeScript

   An Automated Differentiation library (like Pytorch🔥) in Typescript, for educational purposes.

   Very very cool to see!!!

   So yours is SO MUCH MORE fleshed out than mine, but I have a similar (again very fledgling) one going here that uses Typescript:

   https://github.com/Marviel/lab-grad

   I think there are some really neat opportunities to increase the approachability of ML algorithms when you take advantage of the Typescript templating system to explicitly ensure you're not accidentally composing tensors of incorrect shapes.

4. SimSIMD

   3 21 1,286 9.8 C

   Up to 200x Faster Dot Products & Similarity Metrics — for Python, Rust, C, JS, and Swift, supporting f64, f32, f16 real & complex, i8, and bit vectors using SIMD for both AVX2, AVX-512, NEON, SVE, & SVE2 📐

   

5. tfjs

   4 32 18,714 6.0 TypeScript

   A WebGL accelerated JavaScript library for training and deploying ML models.

   

   Many people seem to be unaware of tensorflow.js, an official JS implementation of TF

   https://github.com/tensorflow/tfjs

   I'd love to see PyTorch in JS, but I think unless you get it running on the GPU it won't be able to do much.

6. gpu.js

   5 9 15,190 0.0 JavaScript

   GPU Accelerated JavaScript

   

   You might already be familiar, but a GPU.js backend can provide some speedups via good old WebGL -- no need for WebGPU just yet!

   [0]: https://github.com/gpujs/gpu.js/

7. transformers.js

   6 33 13,200 9.7 JavaScript

   State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server!

   

   BTW: you might want to add support for typed arrays.

   See: https://github.com/xenova/transformers.js/blob/8804c36591d11...

   This is really old, but added as part of the shape of the vector as well: https://github.com/nicolaspanel/numjs/blob/master/src/dtypes...

8. numjs

   7 6 2,420 0.0 JavaScript

   Like NumPy, in JavaScript

   

   BTW: you might want to add support for typed arrays.

   See: https://github.com/xenova/transformers.js/blob/8804c36591d11...

   This is really old, but added as part of the shape of the vector as well: https://github.com/nicolaspanel/numjs/blob/master/src/dtypes...

9. SaaSHub

   www.saashub.com featured

   SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

   

10. onnxruntime

    8 62 15,918 9.9 C++

    ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

    

    tfjs is dead, looking at the commit history. The standard now is to convert PyTorch to onnx, then use onnxruntime (https://github.com/microsoft/onnxruntime/tree/main/js/web) to run the model on the browsdr.

11. onegrad.js

    9 1 0 - JavaScript

    Lightweight deep learning framework in JavaScript

    Cool project! I worked on something similar a while ago to learn about automatic differentiation: https://github.com/hexhowells/onegrad.js

    Needs more layers to be really useful though.

    I always thought examples using browser extensions would be neat since their built in JS and you only need to download the model once.

12. taichi.js

    10 3 485 6.2 TypeScript

    Modern GPU Compute and Rendering in Javascript

    

    FWIW also taichi is quite popular in python and seems has some javascript related implementation (I haven't used it though), taichi.js [0]

    [0] https://github.com/AmesingFlank/taichi.js

13. node-gyp

    11 26 10,120 7.4 Python

    Node.js native addon build tool

    

    The benefit of having it in JS is not speed but portability and access to the JS ecosystem of tools. Having the code run in the browser without needing a complex setup is a huge benefit for sharing demos. Node.js provides a way to use native code as well and it's quite commonly used https://github.com/nodejs/node-gyp so there's no reason you couldn't use those same or similar libraries in a JS implementation.

Suggest a related project