OliverFM / rust_light

A tensor processing library in rust.
MIT License
3 stars 0 forks source link

rust_light

A tensor processing library in rust.

Goals

The main goal of this project is to get a simple tensor processing library working.

  1. Get something that works.
  2. Learn more about rust.
  3. Pybind rust_light
  4. Get rust_light to work on a GPU.

Intended functionality:

  1. implement a Tensor that supports multiplication and additions -- for scalars and matrices/tensors.
  2. Enable full numpy-style broadcasting.
  3. stretch: add some kind of auto-grad.

Contributing:

Contributions welcome! If you'd like to contribute, start by opening an issue.