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