jax.ref module
Contents
jax.ref module#
jax.ref has the API for working withArrayRef.
API#
| Abstract mutable array reference. |
| Mutable array reference. |
| Invalidate a given reference and return its final value. |
| Read a value from an Ref. |
| Create a mutable array reference with initial value |
| Set a value in an Ref in-place. |
| Update an array value inplace while returning the previous value. |
| Add to an element in an Ref in-place. |
Contents
