rocm_jax/docs/jax.experimental.loops.rst
George Necula d24c374d59 An implementation of an experimental syntactic sugar for 'for' loops.
See description in jax/experimental/loops.py.
2019-11-16 17:23:40 +01:00

6 lines
136 B
ReStructuredText

jax.experimental.loops module
=============================
.. automodule:: jax.experimental.loops
:members:
:show-inheritance: