[package]
edition = "2021"
name = "quill-mlx"
version = "0.25.5"
authors = [
"Kyle Kelley <rgbkrk@gmail.com>",
"Minghua Wu <michael.wu1107@gmail.com>",
"David Chavez <david@dcvz.io>",
]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Rust bindings for Apple's MLX machine learning framework, with Metal memory management"
readme = "README.md"
keywords = [
"mlx",
"deep-learning",
"machine-learning",
]
categories = ["science"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/rgbkrk/mlx-rs"
[package.metadata.docs.rs]
targets = [
"aarch64-apple-darwin",
"aarch64-apple-ios",
"aarch64-apple-ios-sim",
]
[features]
accelerate = ["quill-mlx-sys/accelerate"]
default = [
"accelerate",
"metal",
]
metal = ["quill-mlx-sys/metal"]
safetensors = [
"dep:safetensors",
"dep:bytemuck",
]
[lib]
name = "quill_mlx"
path = "src/lib.rs"
[[example]]
name = "linear_regression"
path = "examples/linear_regression.rs"
[[example]]
name = "tutorial"
path = "examples/tutorial.rs"
[dependencies.bytemuck]
version = "1"
features = ["extern_crate_std"]
optional = true
[dependencies.dyn-clone]
version = "1"
[dependencies.half]
version = "2"
[dependencies.itertools]
version = "0.14"
[dependencies.libc]
version = "0.2"
[dependencies.mach-sys]
version = "0.5"
[dependencies.num-complex]
version = "0.4"
[dependencies.num-traits]
version = "0.2"
[dependencies.num_enum]
version = "0.7"
[dependencies.parking_lot]
version = "0.12"
[dependencies.paste]
version = "1"
[dependencies.quill-mlx-internal-macros]
version = "0.25"
[dependencies.quill-mlx-macros]
version = "0.25"
[dependencies.quill-mlx-sys]
version = "=0.2.3"
[dependencies.safetensors]
version = "0.6"
optional = true
[dependencies.smallvec]
version = "1"
[dependencies.strum]
version = "0.27"
features = ["derive"]
[dependencies.thiserror]
version = "2"
[dev-dependencies.float_eq]
version = "1"
[dev-dependencies.pretty_assertions]
version = "1.4.0"
[dev-dependencies.tempfile]
version = "3"