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.8.0 (XLA 9f150f6). Keep in mind that most of them are experimental and don’t depend on their behaviour to be stable between JAX/XLA versions.