Skip to main content

Browse the archive

Revision 737327a42d40ef0ee44bb06c3dad312a7df9a2dc authored by jax authors on 23 September 2022, 19:58:03 UTC, committed by jax authors on 23 September 2022, 19:58:03 UTC 
Merge pull request #12490 from mattjj:improve-leak-checker
 
PiperOrigin-RevId: 476442352
2 parent s a6b24b3 + b6ef90f
Raw File
jax.stages.rst 
jax.stages package
==================

.. automodule:: jax.stages

Classes
-------

.. currentmodule:: jax.stages

.. autoclass:: Wrapped
   :members: lower
   :special-members: __call__

.. autoclass:: Lowered
   :members: in_tree, out_tree, compile, as_text, compiler_ir

.. autoclass:: Compiled
   :members: in_tree, out_tree, as_text, cost_analysis, memory_analysis, runtime_executable
   :special-members: __call__
back to top