meuron 0.4.0

Meuron is a modular neural network library written in rust for training simple neural networks.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use bytemuck::{Pod, Zeroable};
use wgpu;

pub const TENSOR_USAGE: wgpu::BufferUsages = wgpu::BufferUsages::STORAGE
    .union(wgpu::BufferUsages::COPY_SRC)
    .union(wgpu::BufferUsages::COPY_DST);

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct BinopParams {
    pub size: u32,
    pub op: u32,
    pub pad0: u32,
    pub pad1: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct ScalarParams {
    pub size: u32,
    pub op: u32,
    pub scalar: f32,
    pub pad: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct UnaryParams {
    pub size: u32,
    pub op: u32,
    pub pad0: u32,
    pub pad1: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct MatmulDims {
    pub batch: u32,
    pub m: u32,
    pub k: u32,
    pub n: u32,
    pub b_shared: u32,
    pub pad0: u32,
    pub pad1: u32,
    pub pad2: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct BatchDims {
    pub batch: u32,
    pub last_dim: u32,
    pub pad0: u32,
    pub pad1: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct BroadcastDims {
    pub total: u32,
    pub last_dim: u32,
    pub pad0: u32,
    pub pad1: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct SumAxisParams {
    pub outer: u32,
    pub axis_len: u32,
    pub inner: u32,
    pub pad: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct ReduceSumParams {
    pub size: u32,
    pub pad0: u32,
    pub pad1: u32,
    pub pad2: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct GatherParams {
    pub outer: u32,
    pub axis_len: u32,
    pub inner: u32,
    pub n_sel: u32,
}

#[repr(C)]
#[derive(Copy, Clone, Pod, Zeroable)]
pub struct TransposeDims {
    pub ndim: u32,
    pub total: u32,
    pub axis1: u32,
    pub axis2: u32,
    pub out_shape: [u32; 6],
    pub out_stride: [u32; 6],
    pub in_stride: [u32; 6],
    pub pad: [u32; 2],
}