Crate rusty_gradients 

RustyGradients

rusty-gradients - это фреймворк для глубокого обучения на Rust, вдохновленный PyTorch, с фокусом на простоте, производительности и реализации современных архитектур, таких как Трансформеры.

Основной компонент - это Tensor, который поддерживает автоматическое дифференцирование (autograd).

Modules

backend

core

error

Модуль, определяющий типы ошибок для библиотеки RustyGradients.

losses

Модуль, содержащий различные функции потерь (loss functions).

models

nn

Модуль, содержащий строительные блоки для нейронных сетей.

ops

ops_v2

Operations V2 - операции с autograd для TensorV2

optim

Модуль, содержащий трейт Optimizer и его реализации.

serialization

tensor

Модуль, определяющий Tensor — основную структуру данных в библиотеке.

tensor_v2

TensorV2 - Device-agnostic tensor с multi-backend поддержкой

tokenization

wasm_api