burn-jit 0.16.1

Generic backend that can be compiled just-in-time to any shader language target
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

mod base;
mod col2im;
mod direct;
mod gemm;
mod im2col;
mod implicit_gemm;
mod layout_swap;
mod transpose_direct;

mod tune;

pub use base::*;
pub use col2im::*;
pub use direct::*;
pub use gemm::*;
pub use im2col::*;
pub use implicit_gemm::*;
pub use layout_swap::*;
pub use transpose_direct::*;
pub use tune::*;