Crate rustyasg

Crate rustyasg 

Source
Expand description

§RustyASG: Графовый движок для глубокого обучения на Rust

RustyASG — это современный экспериментальный фреймворк для глубокого обучения, написанный на Rust. Его ключевая особенность — архитектура, построенная вокруг Абстрактного Семантического Графа (ASG).

§Пример использования

use std::rc::Rc;
use std::cell::RefCell;
use rustyasg::tensor::{GraphContext, Tensor};
use rustyasg::losses::mse_loss;

// 1. Создаем контекст графа
let context = Rc::new(RefCell::new(GraphContext::new()));

// 2. Определяем символьные входы
let input = Tensor::new_input(&context, "input");
let expected = Tensor::new_input(&context, "expected");

// 3. Строим граф вычислений
let prediction = input.relu(); // Просто пример операции
let loss = mse_loss(&prediction, &expected);

// Граф готов к анализу, дифференцированию и выполнению на бэкенде!

Modules§

analysis
Модуль, содержащий различные “проходы” (passes) для анализа ASG.
asg
Модуль, определяющий ядро Абстрактного Семантического Графа (ASG).
autograd
Модуль для автоматического дифференцирования (Autograd) на основе ASG.
losses
Модуль, содержащий реализации функций потерь в графовой парадигме.
nn
Модуль, содержащий строительные блоки для нейронных сетей.
optimizers
Модуль, содержащий реализации оптимизаторов для обновления весов модели.
runtime
Модуль, содержащий исполнительные среды (бэкенды) для ASG.
tensor
Модуль, определяющий Tensor и GraphContext.