- Initial Condition (IC) Support: All models now require
(q0, p0)as input- New forward signature:
model(V, t, ic)whereic = (batch, 2) ICEmbeddingmodule added for conditioning- Additive conditioning in decoder:
x = x + ic_embed(ic)
- New forward signature:
- Autoregressive Long-term Prediction: Chain multiple windows using predicted endpoints
- New function:
autoregressive_predict(model, V, n_windows, ...)
- New function:
- Data Pipeline Updates:
load_data()now returns(V, t, q, p, ic)with IC extracted fromq[:, 0], p[:, 0]solvers.rsandtrue_trajectories.jluse actual IC from data (not origin)
- Build System:
just build-cargonow builds only required binaries (neural_hamilton, relevant, solvers) - Breaking Change: Existing checkpoints incompatible (new
ic_embedlayers)
- Total data = n * NDIFFCONFIG
- Batch size: 256
- Remove umap
- Introduce
NDIFFCONFIGto control the number of different time configurations per potential.- This version uses
NDIFFCONFIG=2by default, which means two different time configurations per potential.
- This version uses