axonml 0.1.0

A complete ML/AI framework in pure Rust - PyTorch-equivalent functionality
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

# Axonml

A complete, PyTorch-equivalent machine learning framework in pure Rust.

## Features

- **Tensors** - N-dimensional arrays with broadcasting
- **Autograd** - Automatic differentiation
- **Neural Networks** - Linear, Conv, RNN, LSTM, Attention
- **Optimizers** - SGD, Adam, AdamW, RMSprop
- **Data Loading** - Dataset, DataLoader, transforms
- **Vision** - ResNet, VGG, ViT architectures
- **LLM** - BERT, GPT-2 architectures
- **Serialization** - Save/load models, ONNX export
- **Quantization** - INT8/INT4 compression

## Quick Start

```rust
use axonml::prelude::*;

let model = Sequential::new()
    .add(Linear::new(784, 256))
    .add(ReLU)
    .add(Linear::new(256, 10));

let mut optimizer = Adam::new(model.parameters(), 0.001);

for batch in dataloader.iter() {
    let output = model.forward(&batch.data);
    let loss = output.cross_entropy(&batch.targets);
    
    optimizer.zero_grad();
    loss.backward();
    optimizer.step();
}
```

## Documentation

- [GitHub](https://github.com/AutomataNexus/AxonML)
- [API Docs](https://docs.rs/axonml)

## License

MIT OR Apache-2.0