Skip to content

Allow NumPy 2 in dependency metadata - #43

Draft
CharlelieLrt wants to merge 1 commit into
NVlabs:mainfrom
CharlelieLrt:codex/numpy2-metadata
Draft

CharlelieLrt wants to merge 1 commit into
NVlabs:mainfrom
CharlelieLrt:codex/numpy2-metadata

Conversation

@CharlelieLrt

@CharlelieLrt CharlelieLrt commented Oct 14, 2026 •

Copy link
Copy Markdown

Description

This PR removes the numpy<2.0.0 dependency constraint so FastGen can install
alongside packages that require NumPy 2. In particular, it allows the
PhysicsNeMo diffusion-fastgen extra to resolve without weakening PhysicsNeMo's
NumPy dependency.

The change affects dependency metadata only. It does not change training
methods or public APIs, and it remains independent of the loss-hook proposal
in FastGen #44.

Dependencies

  • Depends on

    • This PR starts from FastGen main and has no prerequisite PR.
  • Depends on this PR

    • PhysicsNeMo #2096 adds private adapters and public sCM and MeanFlow losses. Its dependency pin includes this metadata change without waiting for an upstream merge.
  • Later follow-ups

  • Related prerequisites

  • Independent proposal

    • FastGen #44 provides shared loss hooks for external training loops. It does not depend on this metadata change.
  • Conditional follow-up

If FastGen maintainers decline FastGen #44, we would close PhysicsNeMo #2100.
The preceding PhysicsNeMo stack would remain available.

Review process

Reviewers can compare this PR with its immediate parent. After a parent merges,
we update the remaining branches and their PR bases to keep each diff focused.

Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant