Practical HPC Tools Series - High-Performance Computing in JAX
Tuesday, 15 December 2026 -
10:00
Monday, 14 December 2026
Tuesday, 15 December 2026
10:00
Introduction to JAX Concepts and Framework: Introduction to JAX Concepts and Framework
Introduction to JAX Concepts and Framework
10:00 - 10:20
10:20
Exercise and Correction: Exercise and Correction
Exercise and Correction
10:20 - 10:30
10:30
BREAK: BREAK
BREAK
10:30 - 10:45
10:45
More JAX features for numerical computation
10:45 - 11:05
11:05
Exercise and Correction: Exercise and Correction
Exercise and Correction
11:05 - 11:20
11:20
Scaling JAX to multiple nodes: Scaling JAX to multiple nodes
Scaling JAX to multiple nodes
11:20 - 11:35
11:35
Q&A and Wrap up: Q&A and Wrap up
Q&A and Wrap up
11:35 - 12:00