jax_operator.py 7.1 KB