.. py:module:: megengine.jit
.. currentmodule:: megengine.jit

=============
megengine.jit
=============

>>> from megengine.jit import trace

.. autosummary::
   :toctree: api
   :nosignatures:

   trace
   exclude_from_trace
   xla_trace

Trace Config
------------
.. autosummary::
   :toctree: api
   :nosignatures:

   SublinearMemoryConfig
   DTRConfig
   GraphOptimizationConfig