boostr 0.1.0

ML framework built on numr - attention, quantization, model architectures
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

pub mod clock;
pub mod comm;
pub mod gpipe;
pub mod schedule_1f1b;
pub mod schedule_interleaved;
pub mod schedule_zero_bubble;
pub mod stage;

pub use clock::{PipelineAction, PipelineClock};
pub use comm::{recv_activation, send_activation};
pub use gpipe::GpipeSchedule;
pub use schedule_1f1b::{LossFn, PipelineOutput, Schedule1F1B};
pub use schedule_interleaved::ScheduleInterleaved1F1B;
pub use schedule_zero_bubble::ScheduleZeroBubble;
pub use stage::{PipelineStage, StageContext, TrainablePipelineStage, ZeroBubbleStage};