mirror of
https://github.com/ROCm/jax.git
synced 2025-04-26 02:46:09 +00:00
19 lines
242 B
ReStructuredText
19 lines
242 B
ReStructuredText
|
``jax.extend.core`` module
|
||
|
|
==========================
|
||
|
|
|
||
|
|
.. automodule:: jax.extend.core
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: _autosummary
|
||
|
|
|
||
|
|
ClosedJaxpr
|
||
|
|
Jaxpr
|
||
|
|
JaxprEqn
|
||
|
|
Literal
|
||
|
|
Primitive
|
||
|
|
Token
|
||
|
|
Var
|
||
|
|
array_types
|
||
|
|
jaxpr_as_fun
|
||
|
|
primitives
|