exojax icon indicating copy to clipboard operation
exojax copied to clipboard

Published 20 hours ago • verified publisher icon HajimeKawahara

AmClouds is not differentiable yet

Open HajimeKawahara opened this issue 7 months ago • 1 comments

searchsorted strategy in amclouds.compute_cloud_base_pressure_index does not work. I need to think about the implementation of clouds.

    rg_layer, MMRc = amp_nh3.calc_ammodel(
  File "/home/kawahara/exojax/src/exojax/atm/atmphys.py", line 111, in calc_ammodel
    pressure_base = pressures[ibase]
jax.errors.TracerArrayConversionError: The numpy.ndarray conversion method __array__() was called on traced array with shape int32[].
The error occurred while tracing the function update at /home/kawahara/anaconda3/lib/python3.10/site-packages/jaxopt/_src/optax_wrapper.py:121 for jit. This value became a tracer due to JAX operations on these lines:

HajimeKawahara avatar Jul 14 '24 04:07 HajimeKawahara