lumen-io 0.5.0

A tiny ML framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11

import torch
from safetensors.torch import save_file

tensors = {
    "embedding": torch.zeros((2, 2)),
    "attention": torch.rand((3, 3), dtype=torch.float32),
    "bias": torch.tensor([1, 2, 3], dtype=torch.int32) 
}

save_file(tensors, "test.safetensors")