Rust Tensor Compute
This is a personal project to teach myself WebGPU computing, focused
on Machine Learning application.
Features for now:
- Select which GPU to use (if more than 1 in system)
- Clone
- (Batch) Matmul
- Relu
- Transpose
- Fill
- Compare
- Make Contiguous
- [-] Slice (Needs more design/work)
- [-] Index (Needs more design/work)
- Create Views Tensor
In progress:
Element Wise Tensor - Tensor Ops:
- Add
- Sub
- Mul
Scalar - Tensor Ops:
- Add
- Sub
- Mul
Working Example:
Short-term goals
- How should indexing and assignment work?
- Finish public API docs
- More Tests