mlx-macros-burn

Procedural macros for mlx-rs-burn.

Overview

This crate provides public procedural macros for mlx-rs-burn, including:

• #[derive(ModuleParameters)] - Derive macro for neural network module parameters
• Helper macros for defining MLX operations

Installation

This crate is automatically included as a dependency of mlx-rs-burn. You typically don't need to add it directly.

[dependencies]
# Just use mlx-rs-burn instead:
mlx-rs-burn = "0.25.4"

Available Macros

#[derive(ModuleParameters)]

Automatically implements parameter collection for neural network modules:

use mlx_macros::ModuleParameters;

#[derive(ModuleParameters)]
struct MyLayer {
    #[param]
    weights: Array,
    #[param]
    bias: Array,
}

Related Crates

Upstream

Fork of mlx-macros from oxideai/mlx-rs.

License

MIT OR Apache-2.0