Major refactoring for JAX-style classes. - #29

Merged

EstevaoMGomes merged 146 commits into

mainfrom

Aug 7, 2026

Commits on Apr 20, 2025

Add new integrator scripts for field tracing and guiding center dynamics
EstevaoMGomes
committed
871d169
Improve loss functions speed
EstevaoMGomes
committed
f86f5ee

Commits on Apr 21, 2025

Refine integrator analysis
EstevaoMGomes
committed
408c2c5

Commits on Apr 27, 2025

Optimize rotation matrix computation in RotatedCurve function
EstevaoMGomes
committed
6ff79db

Commits on Apr 28, 2025

Edit fo and gc integration analysis plots
EstevaoMGomes
committed
fdb1508
Improve loss functions computational efficiency
EstevaoMGomes
committed
415bbae
Feature: gradient analysis
EstevaoMGomes
committed
f38fdea

Commits on May 4, 2025

Add block_until_ready to integrators analysis
EstevaoMGomes
committed
079cdb7
Minor tweaks to gradient analysis plot
EstevaoMGomes
committed
fe5d81d
Create Poincare Plots analysis
EstevaoMGomes
committed
9595ab4

Commits on May 11, 2025

Commits on May 14, 2025

Commits on May 19, 2025

Commits on May 21, 2025

Commits on May 24, 2025

Commits on May 26, 2025

Fixed energy calculation for fo trajectories
EstevaoMGomes
committed
1c6584c
Finalize gc analysis ESSOS vs SIMSOPT
EstevaoMGomes
committed
32b84ad

Commits on May 28, 2025

Commits on Jun 6, 2025

Based on work from PR #19
EstevaoMGomes
committed
32c36b3
Fix energy calls in integrators analysis
EstevaoMGomes
committed
798731d

Commits on Jun 25, 2025

Commits on Jun 26, 2025

Commits on Jun 27, 2025

Fix Coils.from_ imports
EstevaoMGomes
committed
9c2407b
Finished surface comparison
EstevaoMGomes
committed
8975934

Commits on Jul 9, 2025

Finish simsopt comparison & improve analysis plots
EstevaoMGomes
committed
70dcaa6
Update Ubuntu for workflows
EstevaoMGomes
committed
bbf6cc4

Commits on Sep 8, 2025

Merge branch 'main' into eg/analysis
EstevaoMGomes
committed
e844b67

Commits on Sep 22, 2025

Fixed near-axis & surfaces examples
EstevaoMGomes
committed
7d807ff
Fixed merge changes
EstevaoMGomes
committed
9853449
Deleting comparison_simsopt folder in examples
EstevaoMGomes
committed
1395650

Commits on Oct 14, 2025

Changing surfaces.py to correct the number of modes used, added option to scale the modes with different norms, optimization.py slightly changed to accomodate changes in surfaces. The example optim…
eduardolneto
committed
ddb2282

Commits on Oct 27, 2025

Commits on Oct 28, 2025

Commits on Nov 5, 2025

Fix(surfaces): PyTree able Surfaces
EstevaoMGomes
committed
df5a525
Fix(fields,surfaces): fixed mpol in vmec import
EstevaoMGomes
committed
a4fad74

Commits on Nov 12, 2025

Fix(surfaces): jit and pjit fix
EstevaoMGomes
committed
ce7caf7

Commits on Nov 17, 2025

Feat (losses): Create losses files
EstevaoMGomes
committed
43fd404
Fix (losses): Allow for grad usage during optimization
EstevaoMGomes
committed
ac5c7bc

Commits on Nov 18, 2025

Commits on Dec 9, 2025

refactored examples folder
rogeriojorge
committed
a2af497

Commits on Feb 7, 2026

Feat (example): import optimize_surface_qs from rj/quasisymmetry_surface
EstevaoMGomes
committed
455356e

Commits on Jun 20, 2026

Fix examples for PR #29: API alignment, path fixes, restored loss wrappers, solver kwarg in Tracing, override setters for Curves gamma

Tejas7007
authored and
EstevaoMGomes
committed
39255d5
Address review: restore comments, clarify docstrings, justify processor counts, verify solver differentiability

Tejas7007
authored and
EstevaoMGomes
committed
d22a300
Address review: use SurfaceRZFourier classmethods instead of polymorphic init, revert loss_BdotN kwargs

Tejas7007
authored and
EstevaoMGomes
committed
5ab6d09
Fix solver dropped from Tracing PyTree, clean up loss_BdotN to accept vmec or surface

Tejas7007
authored and
EstevaoMGomes
committed
456168d
Update essos/losses.py

EstevaoMGomes
and
Copilot
authored
f712827

Commits on Jun 26, 2026

Commits on Jun 27, 2026

Commits on Jun 29, 2026

Commits on Jul 1, 2026

Commits on Jul 2, 2026

Commits on Jul 3, 2026

Commits on Jul 4, 2026

Commits on Jul 5, 2026

Addresing unnecessary wrapped code in test_multiobjectives.py origining on PR #21
eduardolneto
committed
b4429c0
Removing breaklines on examples
eduardolneto
committed
5e5bd94
Removing comments on examples
eduardolneto
committed
b02e9f3

Commits on Jul 7, 2026

Commits on Jul 11, 2026

refactor: extracted NearAxis field calculations to standalone pyQSC_JAX package

rathee_agastya
authored and
EstevaoMGomes
committed
0a49f9c
fix: update import paths for standalone pyqsc_jax package

rathee_agastya
authored and
EstevaoMGomes
committed
d268330
fix: add missing BiotSavart import

rathee_agastya
authored and
EstevaoMGomes
committed
3bd9bbf
ADD Dependency for near-axis refactor

rathee_agastya
authored and
EstevaoMGomes
committed
448356f
Adding sfaeguard to coils and surfaces flatten/ubnflatten methods for trace workflow
eduardolneto
committed
22c0c39
Removing repeated cosntructor in from_sismopt methods
eduardolneto
committed
df2b80e
Adding some chanmges
eduardolneto
committed
10cf2e4

Commits on Jul 15, 2026

Commits on Jul 16, 2026

Merge pull request #42 from uwplasma/en/review_eg/examples_fix
eduardolneto
authored
074cf7f

Commits on Jul 17, 2026

Commits on Jul 18, 2026

Fix objective_functions/multiobjectives tests for jitted API (pytree mocks, near-axis shapes matched to pyqsc_jax), fix grad_pytree tracer leak in losses.py

Tejas7007
authored and
EstevaoMGomes
committed
240fe64

Commits on Aug 7, 2026

Adding fixes for ALM and multiobjective tests

eduardolneto
authored and
EstevaoMGomes
committed
91cd376
Fixing ALM test

eduardolneto
authored and
EstevaoMGomes
committed
09fd83c
Fixing ALM test

eduardolneto
authored and
EstevaoMGomes
committed
e3e00d9
Adding extra yaml branches that trigger tests suite at pushes

eduardolneto
authored and
EstevaoMGomes
committed
9593047
Modifying alm and nearaxis related function in tests to reflect source code changes

eduardolneto
authored and
EstevaoMGomes
committed
38bcbb5
Removing unused ALM optimizer methods. These will be reintroduced in later PR for ALM update

eduardolneto
authored and
EstevaoMGomes
committed
f5bfed9
Removing unused ALM optimizer methods. These will be reintroduced in later PR for ALM update

eduardolneto
authored and
EstevaoMGomes
committed
f59866b
Removing unused ALM optimizer methods. These will be reintroduced in later PR for ALM update

eduardolneto
authored and
EstevaoMGomes
committed
295ead2
Fixing test mock parameters from tracers to static in ALM test

eduardolneto
authored and
EstevaoMGomes
committed
50ac788
Changing DummyCurves to mock version of the real object for coil_perturbation test

eduardolneto
authored and
EstevaoMGomes
committed
004d84c
Updating BiotSavart and ALM API compatibility in tests

eduardolneto
authored and
EstevaoMGomes
committed
ef9e57d