Show HN: Glowstick – type level tensor shapes in stable rust
a year ago
- #Tensor
- #Rust
- #Machine Learning
- The crate makes working with tensors in Rust safe, easy, and fun by tracking their shapes in the type system.
- Example usage with candle is provided, including operations like matmul, reshape, unsqueeze, squeeze, narrow, broadcast_add, transpose, conv2d, and flatten.
- The project is currently pre-1.0, meaning breaking changes will be made.
- Features include expressing tensor shapes as types, support for dynamic dimensions, human-readable error messages, and manual checking of type-level shapes.
- Support for all ONNX operations is planned.