[package]
edition = "2024"
rust-version = "1.85"
name = "axonml-train"
version = "0.6.2"
authors = ["AutomataNexus Development Team"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "High-level training, benchmarking, and adversarial training infrastructure for AxonML"
homepage = "https://AutomataNexus.com"
documentation = "https://automatanexus.github.io/AxonML"
readme = "README.md"
keywords = [
"training",
"machine-learning",
"benchmark",
"adversarial",
"deep-learning",
]
categories = [
"science",
"algorithms",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/AutomataNexus/AxonML"
resolver = "2"
[package.metadata.docs.rs]
default-target = "x86_64-unknown-linux-gnu"
targets = ["x86_64-unknown-linux-gnu"]
rustdoc-args = [
"--cfg",
"docsrs",
]
[features]
cuda = [
"axonml-core/cuda",
"axonml-tensor/cuda",
"axonml-nn/cuda",
"axonml-autograd/cuda",
]
default = []
full = [
"vision",
"llm",
]
llm = ["dep:axonml-llm"]
vision = ["dep:axonml-vision"]
[lib]
name = "axonml_train"
path = "src/lib.rs"
[dependencies.axonml-autograd]
version = "0.6.1"
[dependencies.axonml-core]
version = "0.6.1"
[dependencies.axonml-llm]
version = "0.6.1"
optional = true
[dependencies.axonml-nn]
version = "0.6.1"
[dependencies.axonml-optim]
version = "0.6.1"
[dependencies.axonml-tensor]
version = "0.6.1"
[dependencies.axonml-vision]
version = "0.6.1"
optional = true