search

qutip / qutip-jax

JAX backend for QuTiP
BSD 3-Clause "New" or "Revised" License
18 stars 7 forks source link

Update for jax 0.4.23 and dimensions PR #31

Closed Ericgig closed 7 months ago

Ericgig commented 8 months ago

Current jax changed the config import. Also the dimensions PR change the way we need to unflatten Qobj.

Diffrax now raise a warning anytime any array is complex in it's input, so always for us.