4 ms·

I've had my last straw moment with tensorflow some time ago. JAX has been an a pleasure and Im never looking back. It got to a point on my team where it was jus

I've had my last straw moment with tensorflow some time ago. JAX has been an a pleasure and Im never looking back. It got to a point on my team where it was just easier to rewrite our entire distributed training infrastructure in jax with pmap than coerce TF2 into doing what I wanted it to.

jstx1 4y ago
What was TF unable to do that you could do in JAX?
- jdeaton 4y ago
  Reason about my program's behavior.
- dekhn 4y ago
  TF needs to Stop violating the principle of least surprise and spooky action at a distance.