4 ms·

Jax variables are immutable. Code written for CuPy looks similar to numpy but very different from Jax.

• bbminner 2y ago
  Ah, well, that's interesting! Does anyone know how cupy manages tensor mutability?
  • kmaehashi 2y ago
    CuPy tensors (or `ndarray`) provide the same semantics as NumPy. In-place operations are permitted.
• KeplerBoy 2y ago
  Ah yes, stumbled over that recently, but the error message is very helpful and it's a quick change.