Distributed computing for ML is hard. JAX makes it easy (with TPUs & GPUs). Come share your experiences with JAX so we can all learn when JAX is the easy answer and when you should be looking for alternatives.
talk-data.com
S
Speaker
Skye Wanderman-Milne
1
talks
Software Engineer
Google
Filtering by:
Google Cloud Next '25
×
Filter by Event / Source
Talks & appearances
Showing 1 of 1 activities