Viewing a single comment thread. View all comments

anomaly_in_testset t1_itgwh3d wrote

I am primarily a TF 2.x user and have started using Jax as most of my thesis. For me, this is a huge update since TF doesn't support item assignments, numpy like behavior etc. And XLA is a huge update as the speed difference in Jax and TF is significant. So overall, I am very excited.

2

Mefaso t1_itkp4he wrote

>And XLA is a huge update as the speed difference in Jax and TF is significant

Which one is significantly faster? I guess Jax?

2