Show HN: Cant, rust nn lib for learning
9 months ago
- #rust
- #pytorch-like
- #machine-learning
- A machine learning library written in Rust with PyTorch-like functionality.
- Features include automatic differentiation, tensor operations, and GPU acceleration.
- Supports CPU operations using ndarray and Metal GPU acceleration for macOS.
- Includes a PyTorch-like API for ease of use.
- Project setup includes commands for testing, building, formatting, and coverage reports.
- Key components: Tensor struct, computation graph manager, operation definitions, and shape handling.
- Backend support includes CPU (ndarray) and Metal GPU (macOS).
- Testing includes unit tests, model tests, and operation verification.
- Licensed under MIT License.