In place operations in JAX

[jstac]

@mmcky

Can you please search for this line:

In line with immutability, JAX does not support inplace operations:

It looks like JAX is sorting in place now. Could you please evaluate and propose a fix?

[kp992]

lecture-jax/lectures/jax_intro.md

Line 161 in 1a1be32

In line with immutability, JAX does not support inplace operations:

Its here.

[kp992]

For me, its still not inplace:

>>> import jax
>>> jax.__version__
'0.4.23'
>>> import jax.numpy as jnp
>>> a = jnp.array((2, 1))
>>> a.sort()
Array([1, 2], dtype=int32)
>>> a
Array([2, 1], dtype=int32)