ferrotensor-graph 0.2.1

Neural network modules and transformer building blocks for Ferrotensor
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

[package]
name = "ferrotensor-graph"
version = "0.2.1"
edition = "2021"
description = "Neural network modules and transformer building blocks for Ferrotensor"
authors.workspace = true
license.workspace = true
repository.workspace = true
rust-version.workspace = true
homepage = "https://github.com/puranikyashaswin/ferrotensor"
documentation = "https://docs.rs/ferrotensor-graph"
readme = "README.md"
keywords = ["neural-networks", "transformer", "machine-learning", "gpu", "llm"]
categories = ["science", "algorithms"]

[dependencies]
ferrotensor-backend = { path = "../ferrotensor-backend", version = "0.2.1" }
ferrotensor-autograd = { path = "../ferrotensor-autograd", version = "0.2.1" }
ferrotensor-loader = { path = "../ferrotensor-loader", version = "0.2.1" }
env_logger = "0.11"
pollster = "0.3"
log = "0.4"
bytemuck = "1.14"

[dev-dependencies]
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"