Struct MlxClosureCustomJvp 

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

Owned MLX custom JVP closure, released exactly once.

Implementations

impl MlxClosureCustomJvp

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

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

Builds an owned custom-JVP closure from a Rust callable that receives primals, tangents, and the requested output indices.

pub fn empty() -> Self

Creates an empty custom-JVP closure handle.

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

Copies the source closure handle into this handle.

pub fn apply( &self, primals: &MlxArrayVector, tangents: &MlxArrayVector, output_indices: &[i32], ) -> Result<MlxArrayVector, MlxCError>

Applies the custom JVP transform.

Errors

Returns the captured MLX-C description when application fails.

Trait Implementations

impl Debug for MlxClosureCustomJvp

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

Formats the value using the given formatter.

impl Drop for MlxClosureCustomJvp

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.