burn-collective 0.21.0

Backend extension for collective calculations.
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
mod all_reduce;
mod broadcast;
mod reduce;

pub(crate) mod tensor_map;

pub(crate) use all_reduce::*;
pub(crate) use broadcast::*;
pub(crate) use reduce::*;

pub(crate) mod client;
pub(crate) mod server;