Add jaxtyping to requirements.txt for crash sample

[mnabian]

PhysicsNeMo Pull Request

Description

Checklist

• I am familiar with the Contributing Guidelines.
• New or existing tests cover these changes.
• The documentation is up to date with these changes.
• The CHANGELOG.md is up to date with these changes.
• An issue is linked to this pull request.

Dependencies

Review Process

All PRs are reviewed by the PhysicsNeMo team before merging.

Depending on which files are changed, GitHub may automatically assign a maintainer for review.

We are also testing AI-based code review tools (e.g., Greptile), which may add automated comments with a confidence score.
This score reflects the AI’s assessment of merge readiness and is not a qualitative judgment of your work, nor is
it an indication that the PR will be accepted / rejected.

AI-generated feedback should be reviewed critically for usefulness.
You are not required to respond to every AI comment, but they are intended to help both authors and reviewers.
Please react to Greptile comments with 👍 or 👎 to provide feedback on their accuracy.

[mnabian]

/blossom-ci

[greptile-apps]

Greptile Overview

Greptile Summary

Added jaxtyping==0.3.3 as a dependency to the crash sample requirements file.

• The crash sample supports FIGConvUNet model which has a hard dependency on jaxtyping for type annotations
• physicsnemo/models/figconvnet/figconvunet.py:21-25 imports Float from jaxtyping and raises an error if not installed
• The pinned version 0.3.3 ensures compatibility and reproducibility

Confidence Score: 5/5

• This PR is safe to merge with no risk
• This is a straightforward dependency addition with no code changes. The dependency is required for the FIGConvUNet model that is already being used in the crash sample. Without this dependency, users would encounter an ImportError when trying to use the FIGConvUNet model configuration.
• No files require special attention

Important Files Changed

File Analysis

Filename	Score	Overview
examples/structural_mechanics/crash/requirements.txt	5/5	Added jaxtyping==0.3.3 dependency for FIGConvUNet model support

Sequence Diagram

sequenceDiagram
    participant User
    participant CrashSample as Crash Sample
    participant Requirements as requirements.txt
    participant FIGConvUNet as FIGConvUNet Model
    participant jaxtyping as jaxtyping Library
    
    User->>CrashSample: Install dependencies
    CrashSample->>Requirements: Read requirements.txt
    Requirements->>jaxtyping: Install jaxtyping==0.3.3
    jaxtyping-->>Requirements: Installation complete
    
    User->>CrashSample: Run training/inference with FIGConvUNet
    CrashSample->>FIGConvUNet: Import FIGConvUNet
    FIGConvUNet->>jaxtyping: Import Float type annotations
    jaxtyping-->>FIGConvUNet: Type annotations available
    FIGConvUNet-->>CrashSample: Model ready
    CrashSample-->>User: Training/inference proceeds

Loading

_{1 file reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}

[CharlelieLrt]

lgtm