rocm_jax/jax.example_libraries.stax.rst at 120125f3dd6e516e1b7e727b2504182724f697b2 - rocm_jax - Gitea For EOELAB

mirrors/rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00

Roy Frostig 623c201054 [JAX] move example libraries from jax.experimental into jax.example_libraries

The `jax.experimental.stax` and `jax.experimental.optimizers` modules are standalone examples libraries. By contrast, the remaining modules in `jax.experimental` are experimental features of the JAX core system. This change moves the two example libraries, and the README that describes them, to `jax.example_libraries` to reflect this distinction.

PiperOrigin-RevId: 404405186

2021-10-19 17:30:45 -07:00

8 lines

169 B

ReStructuredText

Raw Blame History

 jax.example_libraries.stax module
 =================================
 .. automodule:: jax.example_libraries.stax
     :members:
     :undoc-members:
     :show-inheritance: