Kyle Luther

Different outputs every time fft2 is called

11

### Description Each time fft2 is called, I get a different result. ```python from jax.numpy.fft import fft2 x = np.random.randn(7603, 7603) f1 = fft2(x).block_until_ready() f2 = fft2(x).block_until_ready() f3 = fft2(x).block_until_ready()...

Fix type annotations in render_tiles in warp.py

3

In higher versions of python (>=3.8), you cannot import the `warp.py' module  This error is caused by tuple[int,int] which is not allowed in the function definition, as this is...