ContextQMD
Back to Mlx

Transforms

docs/src/python/transforms.rst

0.31.2262 B
Original Source

.. _transforms:

Transforms

.. currentmodule:: mlx.core

.. autosummary:: :toctree: _autosummary

eval async_eval compile checkpoint custom_function disable_compile enable_compile grad value_and_grad jvp vjp vmap