Static_pages
All XLA Options/Flags
Unfortunately the JAX documentation only seems to list a few common XLA flags. The rest of them is not documented at all outside of the OpenXLA source code. Here I am listing all of them as of JAX/jaxlib 0.10.0 (XLA 187a5eb5). Keep in mind that most of them are experimental and don’t depend on their behaviour to be stable between JAX/XLA versions.
If you are interested in the XLA flags of earlier JAX versions, check out the older versions of this page.