Rory Conlin

Add initial implementation of multigrid optimization

2

Resolves #929

Switch compute from recursive to looped

1

Resolves #1336

### Description I'm using `jax.scipy.sparse.linalg.gmres` with a preconditioner to solve some linear systems, and was trying to figure out why the forward solve would converge just fine but the gradient...

Question/recommendation about versioning

3

I noticed that equinox doesn't pin a maximum version of JAX. On the one hand I get why, since both are nominally alpha/eta/v0.x so compatibility is always kind of a...

Preconditioner for transposed krylov methods

5

I'm using `lx.GMRES` with a preconditioner and trying to understand why the forward solve converges fine but the gradient doesn't. Looking through the source, it seems like the `solver.transpose` throws...

`map_coordinates` and boozer angles

5

It seems like `map_coordinates` doesn't work with boozer coordinates. The use case is computing things on a uniform grid in boozer coordinates. In theory one could write a boozmn file...

make DESC routines to plot eps_effective and Gamma_c

9

I was under the impression that this should work with some sensible defaults for unspecified parameters: ```python eq = desc.examples.get("SOLOVEV") grid = Grid.create_meshgrid([np.array([0.5]), np.linspace(0, 2*np.pi, 20), np.linspace(0, 2*np.pi, 20)], coordinates="raz")...

Add comments about recommeded resolutions to docstrings

2

- [ ] @dpanici can you check that the recommendations for coils seem reasonable? Also if there are other `MagneticField` classes that we can give recommendations for - [ ]...