ENH: `jax_autojit`#284

rgommers merged 5 commits intodata-apis:maindata-apis/array-api-extra:mainfrom

crusaderky:autojitcrusaderky/array-api-extra:autojitCopy head branch name to clipboard