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

//! MLX tensor operations for Burn backend.
//!
//! This module provides implementations of Burn's tensor operation traits
//! for the MLX backend.

mod base;
mod float_ops;
mod int_ops;
mod bool_ops;
mod module_ops;
mod other_ops;

// Re-export operations for internal use
#[allow(unused_imports)]
pub use base::concat;