megengine.jit¶

>>> from megengine.jit import trace

`trace`	Wraps a callable and provide:
`exclude_from_trace`
`xla_trace`	Wraps a callable, and provides accelerated evaluation compiled by xla.

Trace Config¶

`SublinearMemoryConfig`	Configuration for sublinear memory optimization.
`DTRConfig`	Configuration for DTR memory optimization.
`GraphOptimizationConfig`	Configuration for graph optimization: False for OFF, True for ON.

megengine.optimizer.clip_grad_value

trace