rumus 0.3.0

A native-Rust deep learning framework with explicit memory safety and hardware acceleration
Documentation 
1
2
3
4
5
6
7
8
9

// SPDX-License-Identifier: Apache-2.0 OR MIT
pub mod broadcast;
mod core;
mod ops;

pub use self::core::*;
pub use self::ops::{cat, stack};
#[cfg(feature = "gpu")]
pub use self::ops::flash_attention;