[console_scripts]
jax = jax.cli:main
