Struct MlxClosureCustom 

pub struct MlxClosureCustom(/* private fields */);

Owned MLX custom-transform (VJP) closure, released exactly once.

Implementations

impl MlxClosureCustom

pub fn from_fn<F>(callback: F) -> Result<Self, MlxCError>

where F: FnMut(&[MlxArray], &[MlxArray], &[MlxArray]) -> Result<Vec<MlxArray>, MlxCError> + 'static,

Builds an owned custom-transform closure from a Rust callable that receives inputs, outputs, and upstream gradients.

pub fn empty() -> Self

Creates an empty custom-transform closure handle.

pub fn set(&mut self, source: &Self) -> Result<(), MlxCError>

Copies the source closure handle into this handle.

pub fn apply( &self, inputs: &MlxArrayVector, outputs: &MlxArrayVector, upstream_gradients: &MlxArrayVector, ) -> Result<MlxArrayVector, MlxCError>

Applies the custom transform.

Errors

Returns the captured MLX-C description when application fails.

Trait Implementations

impl Debug for MlxClosureCustom

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for MlxClosureCustom

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.