SFU webinar "JAX: a framework for high-performance array computing"
Note: all times are shown in the timezone in which each event occurs.
Date: 16 April 2024 @ 11:00 - 12:00
Timezone: Pacific Daylight Time
Language of instruction: English
Register
Abstract: JAX is an open source framework by Google for high-performance array computations and differentiation. With a NumPy-like API, it uses asynchronous dispatch, just-in-time compilation, and the XLA compiler for linear algebra to run on accelerators (GPUs and TPUs). It is both very fast and flexible. This however comes at the cost of some constraints: JAX requires pure functional programming code. This webinar will introduce JAX's principles and functioning.
Activity log