jax

8dee8fdc - [pallas:mosaic] It is no longer necessary to use "main" as the name of the kernel