[[bench]]
harness = false
name = "monitor_bench"
path = "benches/monitor_bench.rs"
required-features = ["monitor"]
[[bin]]
name = "entrenar"
path = "src/main.rs"
required-features = []
[dependencies.aprender]
version = "0.12.0"
[dependencies.bytemuck]
features = ["derive"]
version = "1.19"
[dependencies.chrono]
optional = true
version = "0.4"
[dependencies.clap]
features = ["derive"]
version = "4.5"
[dependencies.dirs]
version = "5.0"
[dependencies.getrandom]
features = ["wasm_js"]
optional = true
version = "0.3"
[dependencies.hf-hub]
version = "0.4"
[dependencies.memmap2]
version = "0.9"
[dependencies.ndarray]
version = "0.16"
[dependencies.rand]
version = "0.9"
[dependencies.safetensors]
version = "0.7"
[dependencies.serde]
features = ["derive"]
version = "1.0"
[dependencies.serde_json]
version = "1.0"
[dependencies.serde_yaml]
version = "0.9"
[dependencies.thiserror]
version = "2.0"
[dependencies.tokio]
features = ["rt"]
optional = true
version = "1.41"
[dependencies.toml]
version = "0.8"
[dependencies.trueno]
version = "0.7.3"
[dependencies.trueno-db]
optional = true
version = "0.3"
[dependencies.trueno-rag]
optional = true
version = "0.1.0"
[dependencies.trueno-viz]
features = ["terminal"]
version = "0.1.1"
[dependencies.uuid]
features = ["v4", "serde"]
optional = true
version = "1.11"
[dependencies.wasm-bindgen]
optional = true
version = "0.2"
[dev-dependencies.approx]
version = "0.5"
[dev-dependencies.criterion]
version = "0.5"
[dev-dependencies.proptest]
version = "1.4"
[dev-dependencies.tempfile]
version = "3.8"
[[example]]
name = "citl"
path = "examples/citl.rs"
required-features = ["citl"]
[[example]]
name = "distillation"
path = "examples/distillation.rs"
required-features = []
[[example]]
name = "explainability"
path = "examples/explainability.rs"
required-features = []
[[example]]
name = "hf_distillation"
path = "examples/hf_distillation.rs"
required-features = []
[[example]]
name = "llama2-finetune-lora"
path = "examples/llama2/finetune_lora.rs"
required-features = []
[[example]]
name = "llama2-finetune-qlora"
path = "examples/llama2/finetune_qlora.rs"
required-features = []
[[example]]
name = "llama2-memory-benchmarks"
path = "examples/llama2/memory_benchmarks.rs"
required-features = []
[[example]]
name = "llama2-train"
path = "examples/llama2/train.rs"
required-features = []
[[example]]
name = "merge_models"
path = "examples/merge_models.rs"
required-features = []
[[example]]
name = "model_io"
path = "examples/model_io.rs"
required-features = []
[[example]]
name = "monitoring"
path = "examples/monitoring.rs"
required-features = []
[[example]]
name = "train_from_yaml"
path = "examples/train_from_yaml.rs"
required-features = []
[[example]]
name = "train_from_yaml_example"
path = "examples/train_from_yaml_example.rs"
required-features = []
[[example]]
name = "training_loop"
path = "examples/training_loop.rs"
required-features = []
[features]
citl = ["trueno-rag", "tokio", "uuid"]
default = []
gguf = []
monitor = ["trueno-db", "chrono"]
wasm = ["wasm-bindgen", "getrandom"]
[lib]
crate-type = ["cdylib", "rlib"]
name = "entrenar"
path = "src/lib.rs"
[lints.clippy]
assigning_clones = "allow"
case_sensitive_file_extension_comparisons = "allow"
cast_possible_truncation = "allow"
cast_possible_wrap = "allow"
cast_precision_loss = "allow"
cast_sign_loss = "allow"
checked_conversions = "warn"
comparison_chain = "allow"
doc_markdown = "allow"
explicit_deref_methods = "warn"
explicit_iter_loop = "warn"
float_cmp = "allow"
format_push_string = "allow"
implicit_clone = "warn"
inconsistent_struct_constructor = "warn"
inefficient_to_string = "warn"
items_after_statements = "allow"
large_stack_arrays = "allow"
manual_let_else = "allow"
manual_ok_or = "warn"
many_single_char_names = "allow"
match_same_arms = "allow"
match_wildcard_for_single_variants = "allow"
missing_const_for_fn = "allow"
missing_errors_doc = "allow"
missing_panics_doc = "allow"
module_name_repetitions = "allow"
must_use_candidate = "allow"
needless_pass_by_value = "allow"
redundant_closure_for_method_calls = "warn"
return_self_not_must_use = "allow"
similar_names = "allow"
struct_excessive_bools = "allow"
struct_field_names = "allow"
too_many_lines = "allow"
trivially_copy_pass_by_ref = "allow"
unnecessary_wraps = "allow"
unnested_or_patterns = "warn"
unreadable_literal = "allow"
unused_self = "allow"
used_underscore_binding = "allow"
[lints.clippy.all]
level = "warn"
priority = -1
[lints.clippy.pedantic]
level = "warn"
priority = -1
[lints.rust]
missing_debug_implementations = "allow"
trivial_casts = "warn"
trivial_numeric_casts = "warn"
unreachable_pub = "warn"
unsafe_code = "deny"
unused_import_braces = "warn"
unused_lifetimes = "warn"
unused_qualifications = "warn"
[lints.rust.rust_2018_idioms]
level = "warn"
priority = -1
[package]
authors = ["PAIML <hello@paiml.com>"]
autobenches = false
autobins = false
autoexamples = false
autolib = false
autotests = false
build = false
categories = ["science", "mathematics"]
description = "Training & Optimization library with autograd, LoRA, quantization, and model merging"
edition = "2021"
keywords = ["machine-learning", "neural-networks", "autograd", "lora", "quantization"]
license = "MIT"
name = "entrenar"
readme = "README.md"
repository = "https://github.com/paiml/entrenar"
version = "0.2.2"
[profile.dev]
opt-level = 0
[profile.release]
codegen-units = 1
lto = true
opt-level = 3
[[test]]
name = "chaos_llama"
path = "tests/chaos_llama.rs"
[[test]]
name = "gradient_llama"
path = "tests/gradient_llama.rs"
[[test]]
name = "mutation_resistant_llama"
path = "tests/mutation_resistant_llama.rs"
[[test]]
name = "property_llama"
path = "tests/property_llama.rs"