Physicsnemo Shard Tensor
NVIDIA/physicsnemo/skills/physicsnemo-shard-tensor
ai-mlOfficial
Official Provider SkillView repo
Official NVIDIA-authored guidance for PhysicsNeMo ShardTensor domain parallelism — integrate domain parallelism into training/inference scripts (new or existing) with DDP or FSDP2, write and register shard patches to enable new layers/ops, and bootstrap multi-GPU correctness tests. Use when working with ShardTensor, scatter_tensor, domain parallelism, sequence/spatial sharding, ring attention, DeviceMesh + DDP/FSDP2 hybrid parallelism, or physicsnemo.domain_parallel. Do NOT use for generic PyTor
Files8 files
BENCHMARK.md100 lines
Loading editor…
Install
RecommendedOne command — your agent picks it up automatically.
Select an AI agent above to see the install command.
or
Manual Install
More stepsDownload the archive and add the files to your project manually.
Skill details
Versionv1.0.0
AuthorNVIDIA
Categoryai-ml
Skill IDNVIDIA/physicsnemo/skills/physicsnemo-shard-tensor
Files8 files
Related skills
Physicsnemo DiscoverOfficial NVIDIA-authored guidance for navigating PhysicsNeMo — pick the model, datapipe, or example for a SciML/AI4Science task (surrogates, forecasting, downscaling, physics-informed, inverse, generative). Points at existing files via live repo search; never writes code. Do NOT use for installation or environment setup, training-loop or other code authoring/scaffolding, contributor/CI/packaging questions, repo-specific questions in physicsnemo-sym/-cfd/-curator, or general (non-physics) ML/PyToNemoclaw Maintainer Cross Issue SweepFind open issues that a NemoClaw PR may also fix or conflict with. Use when related-issue analysis is requested.Nemoclaw Maintainer Fix E2e FailuresContinuously maintain automatic NemoClaw main E2E results through coordinated repairs. Use for ongoing maintenance, not one-time dispatch or diagnosis.Nemoclaw Maintainer MorningPrepare the NemoClaw morning maintainer plan: triage the backlog, select a target version, and identify release candidates and stragglers.Nemoclaw Maintainer Runtime ProviderImplement or review a native managed NemoClaw runtime provider and its activation or qualification. Excludes the portable experimental profile.Nemoclaw Maintainer TriagePropose and apply authorized Issue Type, Project fields, and labels for NemoClaw issues or PRs, individually or in a batch.