Compile

Optimizes the given model/function.

Abstract Signature:

Compile(model: Callable, fullgraph: bool = False)

PyTorchJAX (Core)NumPyTensorFlowApple MLXFlax NNX

PyTorch

API: torch.compile

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.jit

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.f2py.tests.util.SimplifiedMesonBackend.compile

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.function

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.core.compile

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: nnx.transforms.compilation.Lowered.compile

Strategy: Direct Mapping

Official Docs ↗