[console_scripts]
brax = brax.cli.main:app
