Endpoints — github.com/ml-explore/mlx-examples

CodeHub MCPcopy Index your code

hub / github.com/ml-explore/mlx-examples / endpoints

Endpoints13 in github.com/ml-explore/mlx-examples

⨍ Functions 1,187◇ Types & classes 308↳ Endpoints 13

Routecompute_loss_and_accumulate_grads

partial(mx.compile, inputs=state, outputs=state)

flux/dreambooth.py:None

Routecompute_loss_and_grads

partial(mx.compile, inputs=state, outputs=state)

flux/dreambooth.py:None

partial(mx.compile, inputs=model.state)

mnist/main.py:None

Routegrad_accumulate_and_step

partial(mx.compile, inputs=state, outputs=state)

flux/dreambooth.py:None

Routesingle_step

partial(mx.compile, inputs=state, outputs=state)

flux/dreambooth.py:None

partial(mx.compile, inputs=state, outputs=state)

transformer_lm/main.py:None

partial(mx.compile, inputs=model.state, outputs=model.state)

mnist/main.py:None

partial(mx.compile, inputs=state, outputs=state)

normalizing_flow/main.py:None

partial(mx.compile, inputs=state, outputs=state)

speechcommands/main.py:None

partial(mx.compile, inputs=state, outputs=state)

cifar/main.py:None

partial(mx.compile, inputs=state, outputs=state)

cvae/main.py:None

partial(mx.compile, inputs=state, outputs=state)

gcn/main.py:None

Routetop_k_sampling

partial(mx.compile, inputs=mx.random.state, outputs=mx.random.state)

musicgen/musicgen.py:None