ukiJddlmZddlmZdddededejfdZy) )Any)coreN memory_spaceinit_valrreturnc0tj||S)aCreate a mutable array reference with initial value ``init_val``. For more discussion, see the `Ref guide`_. Args: init_val: A :class:`jax.Array` representing the initial state of the buffer. memory_space: An optional memory space attribute for the Ref. Returns: A :class:`jax.ref.Ref` containing a reference to a mutable buffer. .. _Ref guide: https://docs.jax.dev/en/latest/array_refs.html r)rnew_ref)rrs G/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/ref.pyr r s h\ ::)typingrjax._srcrRefr r r rs037;c;C;488;r