[mosaic_gpu] Fixed unbounded recursion in `FragmentedArray._pointwise`

jax-ml / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

http://jax.readthedocs.io/

Apache License 2.0

30.59k stars 2.82k forks source link

Open copybara-service[bot] opened 4 hours ago

copybara-service[bot] commented 4 hours ago

[mosaic_gpu] Fixed unbounded recursion in FragmentedArray._pointwise