Gregor Thalhammer-Thurner

This issue is to remind me that the code for batched FFTs is still buggy. clFFT supports only batching along a single non-transform axis, but in some cases it is...

Binary operations with Arrays of different memory layout

2

For pyopencl.array.Array I get wrong results for binary operations such as adding if the memory layout differs for both arguments. Say, "A" is a C-contiguous, and "AF" is F-contiguous, then...

Optimization of complex parameters

2

When optimizing a real loss function for a complex variable I encountered some issues with jaxopt 0.7. It seems this use case is less widely used. Are the methods also...

jaxdf.operators.functions.shift_operator shows inconsistent results when applied to FouriereSeries or FiniteDifferences fields. **how to reproduce** At [this gist](https://gist.github.com/geggo/f5122de27753a99b70ec088c2fd6194b) there is a jupyter notebook to demonstrate the behaviour. **Expected behavior** No warnings...