catgrad 0.2.1

a categorical deep learning compiler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

//! Standard library of declarations and definitions for catgrad's `lang` category.

// Actual definitions and the `to_load_ops` helper.
mod stdlib_definitions;
pub use stdlib_definitions::{stdlib, to_load_ops};

// The [`Module`] helper for creating language definitions
mod module;
pub use module::{FnModule, Module};

// Basic ops like reshape, transpose, matmul
pub mod ops;

// Neural network layers (e.g., layernorm, sigmoid, matmul)
pub mod nn;