[JAX] Replace jnp.clip(..., a_min=..., a_max=...) with jnp.clip(..., min=..., max=...).#41

copybara-service[bot] merged 1 commit intomaingoogle/hackable_diffusion:mainfrom

test_890539933google/hackable_diffusion:test_890539933Copy head branch name to clipboard