talk-data.com
Description
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.