xnn 0.2.0

A lightweight ML framework with GPU-first architecture
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

# xnn

[![Crates.io](https://img.shields.io/crates/v/xnn.svg)](https://crates.io/crates/xnn)
[![Documentation](https://docs.rs/xnn/badge.svg)](https://docs.rs/xnn)
[![License](https://img.shields.io/crates/l/xnn.svg)](LICENSE)

A lightweight ML framework built from scratch in Rust with GPU-first architecture.

## Features

- GPU acceleration via [wgpu](https://wgpu.rs/) (Vulkan, Metal, DX12, WebGPU)
- Element types: `f32`, `i32`, `u32`, `bool`
- Cross-platform: Linux, macOS, Windows, Web/WASM
- Automatic compute pipeline caching
- No unsafe code

## Tensor

N-dimensional array with GPU-accelerated operations and automatic broadcasting.

## Elements

| Type | Numeric | Signed | Integer | Float | Logical |
|------|---------|--------|---------|-------|---------|
| `f32` | ✓ | ✓ | | ✓ | |
| `i32` | ✓ | ✓ | ✓ | | |
| `u32` | ✓ | | ✓ | | |
| `bool` | | | | | ✓ |

## Examples

### MNIST Training

Trains a fully-connected network on the MNIST dataset.

```sh
cd examples/mnist-train
cargo run --release
```

### MNIST Web

Interactive digit recognition in the browser using WebGPU and WASM.

```sh
cd examples/mnist-web
wasm-pack build --target web
python3 -m http.server
```

### Linear regression

Trains a simple linear model to fit `y = wx + b`.

```sh
cargo run --release --example linreg
```

### XOR

Trains a 2-layer neural network to solve the XOR problem.

```sh
cargo run --release --example xor
```

## License

MIT — see [LICENSE](LICENSE) for details.