Speakers
Alice Faure
(LUPM)Mr
Jean-Marc Colley
(CNRS/IN2P3/LPNHE)
Description
JAX presentation
The strong points of the JAX, openXLA project
The specificities of JAX array
Transformers and JAX expression
JIT compilation of Python with JAX
Pure functions, concept of static argument and Tracer
An example of vectorization of a function
