Skip to content

Transolver volume #1242

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 44 commits into
base: main
Choose a base branch
from
Open

Transolver volume #1242

Show file tree
Hide file tree
Changes from 43 commits
Commits
Show all changes
44 commits
Select commit Hold shift + click to select a range
6f5b416
Implement transolver ++ physics attention
coreyjadams Oct 2, 2025
dcd0841
Enable ++ in Transolver.
coreyjadams Oct 2, 2025
f8d7736
Merge branch 'NVIDIA:main' into transolver_plus
coreyjadams Oct 3, 2025
58d1325
Merge branch 'NVIDIA:main' into transolver_plus
coreyjadams Oct 6, 2025
3f9d618
Fix temperature correction terms.
coreyjadams Oct 6, 2025
9d76330
Merge branch 'NVIDIA:main' into transolver_plus
coreyjadams Oct 8, 2025
bf82635
Merge branch 'NVIDIA:main' into transolver_plus
coreyjadams Oct 13, 2025
c4ed105
Merge branch 'NVIDIA:main' into transolver_plus
coreyjadams Oct 15, 2025
e97c6d2
Starting work adapting the domino datapipe techniques to transolver.
coreyjadams Oct 22, 2025
987e502
Working towards transolver volume training by mergeing with domino da…
coreyjadams Oct 23, 2025
98d0f76
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Oct 23, 2025
d734cd7
Updating
coreyjadams Oct 23, 2025
606d1ef
Remove printout
coreyjadams Oct 23, 2025
53c72c1
Enable transolver for volumetric data
coreyjadams Oct 23, 2025
9f9b256
Update transolver training script to support either surface or volume…
coreyjadams Oct 27, 2025
713f2a9
Updating datapipe
coreyjadams Oct 30, 2025
908fb52
Tweak transolver volume configs
coreyjadams Oct 30, 2025
9bd97fa
Merge branch 'main' into transolver-volume
coreyjadams Oct 30, 2025
20e00bd
Merge branch 'main' into transolver_plus
coreyjadams Oct 31, 2025
9780219
Merge branch 'NVIDIA:main' into transolver_plus
coreyjadams Oct 31, 2025
27c2cec
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Oct 31, 2025
765dd9e
Add transolverX model
coreyjadams Oct 31, 2025
31df3bd
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Oct 31, 2025
38a500c
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Nov 4, 2025
22af035
Merge branch 'NVIDIA:main' into transolver_plus
coreyjadams Nov 4, 2025
155568e
Enable nearly-uniform sampling of very very large arrays
coreyjadams Nov 4, 2025
af37fd8
limit benchmarking to train epoch, enable profiler in config
coreyjadams Nov 4, 2025
abc86a7
Update volume config slightly
coreyjadams Nov 4, 2025
6860bea
Merge branch 'transolver_plus' into transolver-volume
coreyjadams Nov 4, 2025
926587f
Update training scripts to properly enable data preloading
coreyjadams Nov 4, 2025
6664323
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Nov 5, 2025
1bc4af6
Working towards adding a muon optimzier in transolver
coreyjadams Nov 6, 2025
f7d1739
Add peter's implementation of muon with a combined optimizer. switch…
coreyjadams Nov 7, 2025
56af9dd
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Nov 10, 2025
4c1ebea
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Nov 12, 2025
dc386ef
Merge branch 'NVIDIA:main' into transolver-volume
coreyjadams Nov 17, 2025
c72a6b3
Add updated inference script that can also calculate drag and lift
coreyjadams Nov 17, 2025
a795048
Add better docstrings for typhon
coreyjadams Nov 20, 2025
83ed709
Move typhon to experimental
coreyjadams Nov 20, 2025
d4f2436
Move forwards docstring
coreyjadams Nov 20, 2025
524748e
Adding typhon model and configs.
coreyjadams Nov 20, 2025
751ff68
Update readme.
coreyjadams Nov 20, 2025
dedd507
Merge branch 'main' into transolver-volume
coreyjadams Nov 20, 2025
88dd3be
Merge branch 'main' into transolver-volume
coreyjadams Nov 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
191 changes: 136 additions & 55 deletions examples/cfd/external_aerodynamics/transolver/README.md
View file
Open in desktop

Large diffs are not rendered by default.

195 changes: 0 additions & 195 deletions examples/cfd/external_aerodynamics/transolver/inference_on_zarr.py
View file
Open in desktop

This file was deleted.

88 changes: 0 additions & 88 deletions examples/cfd/external_aerodynamics/transolver/loss.py
View file
Open in desktop

This file was deleted.

2 changes: 1 addition & 1 deletion examples/cfd/external_aerodynamics/transolver/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,5 +5,5 @@ termcolor
torchinfo
einops
transformer_engine[pytorch]
tensorstore
zarr>=3.0
zarrs
Loading