https://github.com/google/jax
Raw File
Tip revision: f09dc2ace1f35aa5117fca2a66e941d1d54fb3ae authored by jax authors on 20 October 2022, 21:41:55 UTC
Clarify in JAX Basics that JAX array creation is also an operation that requires accelerator dispatch and converting to a regular Python type is a blocking operation.
Tip revision: f09dc2a
jax.example_libraries.optimizers.rst
jax.example_libraries.optimizers module
=======================================

.. automodule:: jax.example_libraries.optimizers
    :members:
    :undoc-members:
    :show-inheritance:
back to top