3 ms·
You're right! Maybe we should revise that... I made https://github.com/google/jax/pull/17851 https://github.com/google/jax/pull/17851, comments welcome!
by mattjjatgoogle 3y ago
You're right! Maybe we should revise that... I made https://github.com/google/jax/pull/17851 https://github.com/google/jax/pull/17851, comments welcome!