Closures

typedef struct mlx_closure_ mlx_closure

typedef struct mlx_closure_value_and_grad_ mlx_closure_value_and_grad

typedef struct mlx_closure_custom_ mlx_closure_custom

typedef struct mlx_closure_custom_jvp_ mlx_closure_custom_jvp

typedef struct mlx_closure_custom_vmap_ mlx_closure_custom_vmap

mlx_closure mlx_closure_new()

int mlx_closure_free(mlx_closure cls)

mlx_closure mlx_closure_new_func(int (*fun)(mlx_vector_array*, const mlx_vector_array))

mlx_closure mlx_closure_new_func_payload(int (*fun)(mlx_vector_array*, const mlx_vector_array, void*), void *payload, void (*dtor)(void*))

int mlx_closure_set(mlx_closure *cls, const mlx_closure src)

int mlx_closure_apply(mlx_vector_array *res, mlx_closure cls, const mlx_vector_array input)

mlx_closure mlx_closure_new_unary(int (*fun)(mlx_array*, const mlx_array))

mlx_closure_value_and_grad mlx_closure_value_and_grad_new()

int mlx_closure_value_and_grad_free(mlx_closure_value_and_grad cls)

mlx_closure_value_and_grad mlx_closure_value_and_grad_new_func(int (*fun)(mlx_vector_array*, mlx_vector_array*, const mlx_vector_array))

mlx_closure_value_and_grad mlx_closure_value_and_grad_new_func_payload(int (*fun)(mlx_vector_array*, mlx_vector_array*, const mlx_vector_array, void*), void *payload, void (*dtor)(void*))

int mlx_closure_value_and_grad_set(mlx_closure_value_and_grad *cls, const mlx_closure_value_and_grad src)

int mlx_closure_value_and_grad_apply(mlx_vector_array *res_0, mlx_vector_array *res_1, mlx_closure_value_and_grad cls, const mlx_vector_array input)

mlx_closure_custom mlx_closure_custom_new()

int mlx_closure_custom_free(mlx_closure_custom cls)

mlx_closure_custom mlx_closure_custom_new_func(int (*fun)(mlx_vector_array*, const mlx_vector_array, const mlx_vector_array, const mlx_vector_array))

mlx_closure_custom mlx_closure_custom_new_func_payload(int (*fun)(mlx_vector_array*, const mlx_vector_array, const mlx_vector_array, const mlx_vector_array, void*), void *payload, void (*dtor)(void*))

int mlx_closure_custom_set(mlx_closure_custom *cls, const mlx_closure_custom src)

int mlx_closure_custom_apply(mlx_vector_array *res, mlx_closure_custom cls, const mlx_vector_array input_0, const mlx_vector_array input_1, const mlx_vector_array input_2)

mlx_closure_custom_jvp mlx_closure_custom_jvp_new()

int mlx_closure_custom_jvp_free(mlx_closure_custom_jvp cls)

mlx_closure_custom_jvp mlx_closure_custom_jvp_new_func(int (*fun)(mlx_vector_array*, const mlx_vector_array, const mlx_vector_array, const int*, size_t _num))

mlx_closure_custom_jvp mlx_closure_custom_jvp_new_func_payload(int (*fun)(mlx_vector_array*, const mlx_vector_array, const mlx_vector_array, const int*, size_t _num, void*), void *payload, void (*dtor)(void*))

int mlx_closure_custom_jvp_set(mlx_closure_custom_jvp *cls, const mlx_closure_custom_jvp src)

int mlx_closure_custom_jvp_apply(mlx_vector_array *res, mlx_closure_custom_jvp cls, const mlx_vector_array input_0, const mlx_vector_array input_1, const int *input_2, size_t input_2_num)

mlx_closure_custom_vmap mlx_closure_custom_vmap_new()

int mlx_closure_custom_vmap_free(mlx_closure_custom_vmap cls)

mlx_closure_custom_vmap mlx_closure_custom_vmap_new_func(int (*fun)(mlx_vector_array*, mlx_vector_int*, const mlx_vector_array, const int*, size_t _num))

mlx_closure_custom_vmap mlx_closure_custom_vmap_new_func_payload(int (*fun)(mlx_vector_array*, mlx_vector_int*, const mlx_vector_array, const int*, size_t _num, void*), void *payload, void (*dtor)(void*))

int mlx_closure_custom_vmap_set(mlx_closure_custom_vmap *cls, const mlx_closure_custom_vmap src)

int mlx_closure_custom_vmap_apply(mlx_vector_array *res_0, mlx_vector_int *res_1, mlx_closure_custom_vmap cls, const mlx_vector_array input_0, const int *input_1, size_t input_1_num)

struct mlx_closure_

#include <closure.h>

struct mlx_closure_value_and_grad_

#include <closure.h>

struct mlx_closure_custom_

#include <closure.h>

struct mlx_closure_custom_jvp_

#include <closure.h>

struct mlx_closure_custom_vmap_

#include <closure.h>