webgpu-tensors

Experimental WebGPU/Js Tensors library. Inspired by Pytorch.

Run examples

• Example 0 : just a tensors copy (Rust native implementation).
• Example 1 : several tests.
• Example 2 : Create a simple 2-layer neural network (Typescript and Pytorch implementation to compare results)
• Example 3 : Import a csv dataset (TODO)
• Example x : Transformer (TODO)

Using Deno

deno run ./examples/ex1.ts

Using Vite

npm run dev

It will run on your browser using WebGPU if present, otherwise CPU.

Using Rust native

cargo run --example ex0

Using Python

Pytorch reference

python examples/ex2.py

Run tests

Using Deno

deno task test

Deno coverage is not working: Issue

Using Vite

npm run test

No WebGPU on Node.js

Using Python

Pytorch reference

python -m unittest tests/tensors_test.py