mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +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
|