Radiation transport example#1647

Open

melo-gonzo wants to merge 69 commits into

NVIDIA:mainfrom

melo-gonzo:radiation-transport-example

Commits on May 14, 2026

feat: consolidated radiation transport surrogate model training/eval recipe
melo-gonzo
committed
feat: update eval plots

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: update checkpointing logic

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: checkpointing updates

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: checkpointing updates

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: checkpoint updates

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: checkpoint logic

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: skip logging nan/inf checkpoints

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: requiring split file

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: default to top_model for inference

Carmelo Gonzales
authored and
melo-gonzo
committed
fix: eval sampler removing duplicates

Carmelo Gonzales
authored and
melo-gonzo
committed
fix: strict checking for physics loss required keys

Carmelo Gonzales
authored and
melo-gonzo
committed
fix: validatio metric accregation and checkpointing updates

Carmelo Gonzales
authored and
melo-gonzo
committed
fix: explicit padding to flux

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: cleaning time dependent training references

Carmelo Gonzales
authored and
melo-gonzo
committed
docs: update readme

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: purging some unused code

Carmelo Gonzales
authored and
melo-gonzo
committed
feat: purging unused code
melo-gonzo
committed
feat: purging unused code
melo-gonzo
committed
feat: some more refactoring and purging
melo-gonzo
committed
feat: some more refactoring and purging
melo-gonzo
committed
feat: a few fixes and cleaning up nits
melo-gonzo
committed
fix: forcing single process inference

Carmelo Gonzales
authored and
melo-gonzo
committed
fix: pre-commit
melo-gonzo
committed
refactor: porting to pmsh
melo-gonzo
committed
docs: refresh readme
melo-gonzo
committed
refactor: read geometry info from sidecar json
melo-gonzo
committed
fix: rename zarr to mesh
melo-gonzo
committed
fix: inference flux stats required or infered from hydra
melo-gonzo
committed
fix: purging zarr refs, update some npy->torch, remove legacy comments
melo-gonzo
committed
fix: cleaning some old comments, imports
melo-gonzo
committed
refactor: use only one metric for checkpointing
melo-gonzo
committed
refactor: clearing out stale code, comments, and duplicate caching
melo-gonzo
committed
refactor: inference script cleanup - moving plotting to standalone file, cleaning up checkpoint loading, metrics calculations
melo-gonzo
committed
refactor: cleaning out stale transforms code and consolidating upstream updates
melo-gonzo
committed
refactor: consolidating loss calculations, pinning one scheduler, removing stale code
melo-gonzo
committed
refactor: consolidating train/trainer logic
melo-gonzo
committed
refactor: cleaning out stale transforms code, adding materials utils
melo-gonzo
committed
fix: purging some excessive comments
melo-gonzo
committed
refactor: delete material file
melo-gonzo
committed
refactor: updating configs per refactors
melo-gonzo
committed
feat: adding eval metrics file
melo-gonzo
committed
feat: adding qoi file
melo-gonzo
committed
feat: add visualizations file
melo-gonzo
committed
fix: path update
melo-gonzo
committed
fix: rename steady state to final time
melo-gonzo
committed
docs: update readme QoI
melo-gonzo
committed
fix: update example location
melo-gonzo
committed
docs: update readem
melo-gonzo
committed
docs: update changelog with RTE example
melo-gonzo
committed
fix: removing unused args
melo-gonzo
committed
docs: add split file to args for training
melo-gonzo
committed
fix: dropping torch.backends.cudnn.benchmark = True
melo-gonzo
committed
fix: move grad norm to configurable arg
melo-gonzo
committed
docs: add note for grad norm
melo-gonzo
committed
fix: updating the parse_amp fn
melo-gonzo
committed
fix: val loss tracking metadata bug
melo-gonzo
committed

Commits on May 15, 2026

Commits on May 18, 2026

Commits on May 19, 2026

feat: update dataset to work with updated pmsh
melo-gonzo
committed

Commits on May 22, 2026

fix: deterministic loading of samples from file lists
melo-gonzo
committed