WHY JAX? Why the Hell a 3rd ML framework in 2023?
YouTube Viewers YouTube Viewers
32.1K subscribers
11,137 views
0

 Published On Nov 13, 2022

100x faster on GPU. Why do I need a 3rd Machine Learning Framework, next to TensorFlow2 and PyTorch? Automatically scalable Cloud infrastructure (TPU Pods v4) on Google Cloud and Nvidia H100 SuperPods (... AWS) with XLA compiler are fast! Really fast!

Based on my last video ( ) we look at a new powerful ML framework from the cloud infrastructure side. Primary factors are SPEED, more SPEED, SCALE, Memory and COSTS ($$) for ML training and inference at cloud supercomputer.

#ai
#datascience
#pytorch
#tensorflow2
#jax
#machinelearning

show more

Share/Embed