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