Dario Panici

some TODO items I see as needed for this PR: - [ ] Update the objective functions test as I had pointed out, so that the tests pass - [...

running the command `make html` in the `docs` folder should make the docs locally and you can view them in the `docs/_build` folder (I think) but you can a so...

@f0uriest what is in rc/particles? what would the order of operations be for getting this done, would it be rc/particles, then retrofitting #734 and #1410 to use the API defined...

What needs to be done for this? something like allowing ProximalProjection to accept nonlinear constraints other than the eq ones, or just wrapping the nonlinear constraints that are not eq...

@f0uriest bump on my question?

@unalmis check what is done and check boxes

> As discussed in [#1508](https://github.com/PlasmaControl/DESC/pull/1508) indexing operations are bottlenecks for the transforms. Below is brief recap and my suggested solution that will also greatly reduce/simplify DESC code. > > 1....

After #1688 , we will have some in-place operations which may not actually be in-place unless the optimize functions are placed under `jit`, the files affected are https://github.com/PlasmaControl/DESC/blob/master/desc/optimize/fmin_scalar.py https://github.com/PlasmaControl/DESC/blob/master/desc/optimize/least_squares.py https://github.com/PlasmaControl/DESC/blob/master/desc/optimize/aug_lagrangian.py...

I think this PR needs to be in to fix the `test_lstsq` test thats failing on master rn (sorry I definitely forgot that that was something we had to wait...

> > It also looks like it doesn't significantly decrease the pytree complexity, and from the profiling it also seems like the pytree operations aren't a significant chunk of the...