Nvalchemi Training Api
NVIDIA/nvalchemi-toolkit/.claude/skills/nvalchemi-training-api
productivityOfficial
Official Provider SkillView repo
How to configure nvalchemi training workflows with TrainingStrategy, custom training functions, standalone or composed losses, loss-weight schedules, optimizer and scheduler configs, validation, hooks, restartable checkpoints, model-agnostic inputs, and scaling to multiple GPUs or nodes with DistributedManager and DDPHook. Use when training a model from scratch, setting up optimizers, schedulers, validation, or checkpointing, or scaling a run across GPUs or nodes (DDP); for adapting a pretrained
Files1 files
SKILL.md282 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 file and paste it into your agent's system prompt.
Skill details
Versionv1.0.0
AuthorNVIDIA
Categoryproductivity
Skill IDNVIDIA/nvalchemi-toolkit/.claude/skills/nvalchemi-training-api
Related skills
Nvalchemi Data StorageHow to write, read, compose, and load atomic data using nvalchemi's composable Zarr-backed storage pipeline (Writer, Reader, Dataset, MultiDataset, DataLoader). Use when saving simulation outputs or trajectories to disk, converting structures (e.g. ASE / extxyz) into Zarr stores, assembling datasets for training or inference, or wiring a DataLoader to stream batches to the GPU.Nvalchemi Data StructuresHow to use AtomicData and Batch, the core graph-based data structures for representing atomic systems and batching them for GPU computation. Use when building systems from positions, cells, and atomic numbers, converting from ASE Atoms, batching or unbatching structures, reading per-atom vs per-graph tensors, grouping graphs within a batch (e.g. NEB path images or ensemble members) via group_idx/GroupLayout, or debugging shape, dtype, or device errors in model inputs.Nvalchemi DistillationHow to distill a large teacher MLIP into a small student with DistillationStrategy — teacher signals and offline dataset labeling, the teacher_* loss targets, the on-policy segment loop (propagator, replay buffer, mixed loader), checkpoints and restart, and accuracy/stability evaluation with acceptance thresholds. Use when training a small student to reproduce a big model's energies, forces, stress, or per-atom energies, generating training frames from the student's own trajectories, or gating aNvalchemi DistributedHow to run domain-decomposed (multi-GPU) MLIP simulations with DomainParallel — choose between the halo and graph-partition strategies, author a distribution_spec so a bring-your-own model runs under domain decomposition, and write a custom dynamics integrator that stays correct across ranks.Nvalchemi Dynamics ApiHow to configure and run dynamics simulations, compose multi-stage pipelines (FusedStage, DistributedPipeline), use inflight batching, and manage data sinks. Use when writing any simulation script — molecular dynamics (NVE/NVT), structure relaxation or geometry optimization (e.g. FIRE2), equation-of-state or adsorption scans — or orchestrating many structures through a batched GPU pipeline.Nvalchemi Dynamics HooksHow to use and write dynamics hooks — callbacks that observe or modify batch state at specific points during each simulation step. Use when a simulation needs neighbor-list rebuilds, convergence checks or early stopping, temperature control, per-step logging or trajectory capture, or any custom per-step behavior attached to a dynamics run.