Add an independent AB-UPT operator and unified aero recipe

Open
#2,162 0 comments 0 reactions 0 assignees View on GitHub

Maintainers usually reply within 2 days

@mnabian is already working on this.

Since Oct 23, 2026.

  • #2163 by @mnabian — open
  • #2164 by @mnabian — open

Assessment

Difficulty
5/5
Estimated time
Over a week
Newbie friendliness
25/100
Issue type
Feature
Clarity
Mostly clear
Activity status
Active
Tech stack
python, pytorch

Research direction

Start by reading arXiv:2502.09692v4 and inspect the existing PhysicsNeMo components that support the requested model and aerodynamics recipe. The scope includes model implementation, training modes, sampling, checkpoint round trips, and physical-field export; done means these areas are covered without claiming numerical parity or benchmark reproduction.

Written by the indexing model from the issue text.

Description

Implement the Anchored-Branched Universal Physics Transformer from arXiv:2502.09692v4, using the paper and existing PhysicsNeMo components without research implementation source code or weights.

Scope:

  • Experimental operator with geometry supernode pooling, interacting surface/volume anchors, reusable query contexts, and a physically scaled finite-difference vorticity helper.
  • Separate model and unified external aerodynamics recipe PRs.
  • Surface-only, volume-only, and joint training with unequal query counts, reproducible sampling, checkpoint round trips, and physical-field export.

This is an independent implementation, not a claim of numerical parity or reproduction of benchmark accuracy.

Dominant language
Python
Stars
3.4k
Forks
828
Avg merge
4d 4h
Merged PRs (30d)
78

Getting set up

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from NVIDIA/physicsnemo

All issues in NVIDIA/physicsnemo

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.