Struct MlxClosureValueAndGrad 

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

Owned MLX value-and-gradient closure, released exactly once.

Implementations

impl MlxClosureValueAndGrad

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

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

Builds an owned value-and-gradient closure from a Rust callable returning both outputs and gradients.

pub fn empty() -> Self

Creates an empty value-and-gradient closure handle.

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

Copies the source closure handle into this handle.

pub fn apply( &self, input: &MlxArrayVector, ) -> Result<(MlxArrayVector, MlxArrayVector), MlxCError>

Applies the closure, producing both the outputs and the gradients.

Errors

Returns the captured MLX-C description when application fails.

Trait Implementations

impl Debug for MlxClosureValueAndGrad

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

Formats the value using the given formatter.

impl Drop for MlxClosureValueAndGrad

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.