Gemini models are written in Jax which through the XLA compiler can be compiled ...

		drdirk 10 months ago \| parent \| context \| favorite \| on: Google will let companies run Gemini models in the... Gemini models are written in Jax which through the XLA compiler can be compiled either to TPU or GPU hardware. Performance may differ but Google (and Nvidia) are very interested in having good performance on both platforms.