burn-mlx 0.1.2

MLX backend for Burn deep learning framework - native Apple Silicon GPU acceleration
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

[dependencies.burn]
default-features = false
features = ["std"]
version = "0.16"

[dependencies.burn-tensor]
version = "0.16"

[dependencies.derive-new]
version = "0.7"

[dependencies.half]
features = ["num-traits"]
version = "2.4"

[dependencies.mlx-rs]
package = "mlx-rs-burn"
version = "0.25.5"

[dependencies.num-traits]
version = "0.2"

[dev-dependencies.burn]
features = ["train", "dataset"]
version = "0.16"

[dev-dependencies.serial_test]
version = "3.2"

[features]
autotune = []
default = []
fusion = []

[lib]
name = "burn_mlx"
path = "src/lib.rs"

[package]
authors = ["TuringWorks"]
autobenches = false
autobins = false
autoexamples = false
autolib = false
autotests = false
build = false
categories = ["science", "algorithms"]
description = "MLX backend for Burn deep learning framework - native Apple Silicon GPU acceleration"
documentation = "https://github.com/TuringWorks/burn-mlx#readme"
edition = "2021"
homepage = "https://github.com/TuringWorks/burn-mlx"
keywords = ["burn", "mlx", "apple-silicon", "machine-learning", "gpu"]
license = "Apache-2.0"
name = "burn-mlx"
readme = "README.md"
repository = "https://github.com/TuringWorks/burn-mlx"
version = "0.1.2"

[package.metadata.docs.rs]
rustdoc-args = ["--cfg", "docsrs"]
targets = ["aarch64-apple-darwin"]