mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
12 lines
205 B
ReStructuredText
12 lines
205 B
ReStructuredText
jax.experimental package
|
|
========================
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
jax.experimental.optimizers
|
|
jax.experimental.stax
|
|
jax.experimental.vectorize
|
|
|
|
.. automodule:: jax.experimental
|