burn-backend 0.22.0-pre.3

Core backend interfaces and data structures for executing tensor operations in Burn.
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
mod base;
mod device;
mod memory_pools;
mod primitive;

pub use base::*;
pub use device::*;
pub use memory_pools::*;
pub use primitive::*;

/// Backend operations on tensors.
pub mod ops;

/// Distributed backend extension.
pub mod distributed;