forge-ml 0.1.0

A WebGPU-native machine learning runtime in Rust: train and run GPT-2 on any GPU wgpu reaches, with no CUDA and no Python
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

{
  "backend": "WebGPU: NVIDIA RTX A5000 (Vulkan)",
  "ids": [
    30,
    27,
    25,
    17,
    27,
    10,
    0,
    21,
    1,
    61,
    47,
    50,
    50,
    1,
    52,
    53,
    58,
    1,
    40,
    43,
    1,
    57,
    53,
    1,
    57,
    53,
    1,
    44,
    39,
    56,
    1,
    39,
    57,
    1,
    58,
    46
  ],
  "max_new_tokens": 30,
  "model": "assets/shakespeare_char",
  "prompt": "ROMEO:"
}