burn-mlx/Cargo.toml at main · TuringWorks/burn-mlx · GitHub

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
28
29
30
31
32
33
34
35
36
37
38
[package]
name = "burn-mlx"
version = "0.1.2"
edition = "2021"
authors = ["TuringWorks"]
description = "MLX backend for Burn deep learning framework - native Apple Silicon GPU acceleration"
license = "Apache-2.0"
repository = "https://github.com/TuringWorks/burn-mlx"
homepage = "https://github.com/TuringWorks/burn-mlx"
documentation = "https://github.com/TuringWorks/burn-mlx#readme"
readme = "README.md"
keywords = ["burn", "mlx", "apple-silicon", "machine-learning", "gpu"]
categories = ["science", "algorithms"]

[package.metadata.docs.rs]
# docs.rs cannot build this crate (requires macOS + Apple Silicon)
# Point users to the README for documentation
targets = ["aarch64-apple-darwin"]
rustdoc-args = ["--cfg", "docsrs"]

[features]
default = []
# Fusion optimization (combining multiple operations)
fusion = []
# Autodiff support
autotune = []

[dependencies]
burn = { version = "0.16", default-features = false, features = ["std"] }
burn-tensor = "0.16"
mlx-rs = { package = "mlx-rs-burn", version = "0.25.5" }
derive-new = "0.7"
half = { version = "2.4", features = ["num-traits"] }
num-traits = "0.2"

[dev-dependencies]
burn = { version = "0.16", features = ["train", "dataset"] }
serial_test = "3.2"