Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add
compiler_options
argument to jax.jit
.
This exists on `Compiled` object via AOT too i.e. `jit(f).lower(*args).compile(compiler_options={})` PiperOrigin-RevId: 692283964 Change-Id: I7d2431b3b372305d574e0c0322a85e027da55ec6
- Loading branch information