nvalchemi-toolkit-ops
High-performance NVIDIA Warp primitives for GPU-enabled computational chemistry and atomistic simulation workflows.
Decision gist · record as of 2026-08-14
Yes, if you are working on GPU-accelerated atomistic simulations or molecular dynamics and need production-ready, batched kernels. The package is actively maintained, has no known vulnerabilities, carries a permissive Apache-2.0 license, and integrates cleanly with PyTorch and JAX. Install friction is low. The main constraint is the requirement for CUDA 12 or 13 and Python 3.11–3.14; verify CPU performance expectations if GPU access is limited.AI-flagged interpretation of the facts on this page — verify before relying
Before you install
- Requires CUDA 12 or 13 and Python 3.11–3.14.
- GPU device recommended for performance; CPU execution support and performance characteristics require verification.
- Low friction: pure Python wheel with only numpy and warp-lang as runtime dependencies.
License · maintenance · safety
Apache-2.0 (permissive) — Apache-2.0 (permissive): you may use, modify, and distribute this package freely in commercial and private projects, provided you retain the license notice and do not hold the authors liable.
last release 2026-08-04 (10 days)
0 known vulnerabilities (OSV.dev, 2026-08-14) · 238,443 downloads/mo, #8,938 on PyPI
Alternatives
Verify before relying
pip install nvalchemi-toolkit-ops
from nvalchemiops.torch.neighbors import neighbor_list
positions = ... # [num_atoms, 3]
cell = ... # [num_systems, 3, 3]
pbc = ... # [num_systems, 3]
edge_index, neighbor_ptr, shifts = neighbor_list(
positions, cutoff=6.0, cell=cell, pbc=pbc, return_neighbor_list=True
)- Whether CPU execution is supported and at what performance penalty relative to GPU.
- Exact performance scaling limits and typical throughput on current GPU hardware.
- Whether JAX bindings are feature-complete parity with PyTorch or have limitations.
- Specific CUDA 12 vs. CUDA 13 compatibility and fallback behavior.
What it is and what it does
NVIDIA ALCHEMI Toolkit-Ops is a collection of GPU-optimized kernels written in warp-lang for accelerating atomistic simulations. It provides batched, high-throughput primitives for neighbor list computation (naive, cell-list, and tiled cluster-pair methods), molecular dynamics (NVE, NVT, NPT, NPH ensembles with multiple thermostat options), geometry optimization (FIRE and FIRE2), and interatomic interactions including DFT-D3 dispersion and electrostatics (DSF, Ewald, PME). The package targets systems with large atom counts and microsecond-scale per-atom throughput on GPUs.
It integrates with PyTorch and JAX, enabling differentiable computation of forces, charge gradients, virials, and stress tensors. The kernels are modular and reusable, intended for library developers, researchers developing new methods, and engineers building production molecular dynamics or interatomic potential applications. Installation is straightforward (pure Python wheel with numpy and warp-lang dependencies), but requires PyTorch or JAX with CUDA support and Python 3.11–3.14.
Use it for
- Accelerate existing molecular dynamics workflows by replacing CPU neighbor list computation with GPU-batched cell-list or cluster-pair methods.
- Compute DFT-D3 dispersion corrections on batches of molecules during model training with automatic differentiation.
- Build production molecular dynamics simulations with NVT or NPT ensembles using GPU-optimized Langevin or Nosé-Hoover thermostats.
- Perform geometry optimization (lattice and coordinate relaxation) via FIRE2 on large systems without implementing custom GPU kernels.
- Evaluate particle mesh Ewald electrostatics with automatic parameter tuning and backpropagation for charge-aware training.
Worth the install?
AI-flagged interpretation of the facts on this page. Verify before relying on it.
Yes, if you are working on GPU-accelerated atomistic simulations or molecular dynamics and need production-ready, batched kernels.
The package is actively maintained, has no known vulnerabilities, carries a permissive Apache-2.0 license, and integrates cleanly with PyTorch and JAX. Install friction is low. The main constraint is the requirement for CUDA 12 or 13 and Python 3.11–3.14; verify CPU performance expectations if GPU access is limited.
Install
nvalchemi-toolkit-ops on PyPI
Before you install
Low friction: pure Python wheel with only numpy and warp-lang as runtime dependencies. Active maintenance (released 10 days ago). Requires Python 3.11–3.14 and CUDA 12 or 13; no system library dependencies beyond what the GPU framework already requires.
Requires CUDA 12 or 13 and Python 3.11–3.14. GPU device recommended for performance; CPU execution support and performance characteristics require verification.
License in practice
Apache-2.0 (permissive): you may use, modify, and distribute this package freely in commercial and private projects, provided you retain the license notice and do not hold the authors liable.
Quickstart
pip install nvalchemi-toolkit-ops
from nvalchemiops.torch.neighbors import neighbor_list
positions = ... # [num_atoms, 3]
cell = ... # [num_systems, 3, 3]
pbc = ... # [num_systems, 3]
edge_index, neighbor_ptr, shifts = neighbor_list(
positions, cutoff=6.0, cell=cell, pbc=pbc, return_neighbor_list=True
)Verify before relying
- Whether CPU execution is supported and at what performance penalty relative to GPU.
- Exact performance scaling limits and typical throughput on current GPU hardware.
- Whether JAX bindings are feature-complete parity with PyTorch or have limitations.
- Specific CUDA 12 vs. CUDA 13 compatibility and fallback behavior.
Package facts
| License | Apache-2.0 permissive |
| Python support | Supports the current Python release <3.15,>=3.11 |
| Install friction | Low. Pure-Python wheel |
| Runtime dependencies | 2 packagesnumpywarp-lang |
| Maintenance | Actively maintained 10 days since the last release |
| First released | |
| Downloads | 238,443 / month, #8,938 on PyPI 30-day window, as of 2026-08-14 |
| Known vulnerabilities | None known OSV.dev, checked 2026-08-14 |
| Classifiers | Environment :: GPUIntended Audience :: Science/ResearchLicense :: OSI Approved :: Apache Software LicenseOperating System :: OS IndependentProgramming Language :: Python :: 3Topic :: Scientific/Engineering :: ChemistryTopic :: Scientific/Engineering :: Physics |
Evidence: nvalchemi_toolkit_ops-0.4.1-py3-none-any.whl
Tags
Let your AI agent find packages like this
Example. Real query, live index.
You found this page by searching. An agent finds it by wishing: SkillFed indexes 14,416 PyPI packages by what they can do, searchable in plain language.
wish › “gpu molecular dynamics”
- nvalchemi-toolkit-opsGPU-accelerated batched primitives for atomistic simulation: neighbor…
- OpenMMOpenMM is a Python wrapper for a C++ molecular dynamics simulation…
- vesin-torchComputes neighbor lists for atomistic systems efficiently, with…
Give your agent the search over MCP, or paste the wish link into any chat.
More Physics packages
NetworkX provides data structures and algorithms for creating, analyzing, and manipulating graphs and networks, supporting everything from simple undirected graphs to complex directed and weighted networks.
SymPy is a Python library for symbolic mathematics, performing algebraic manipulation, calculus, equation solving, and mathematical expression simplification without numerical approximation.
Pydicom reads, modifies, and writes DICOM medical imaging files in pure Python, with optional NumPy support for pixel data as arrays.
Install it if you work with medical imaging data, DICOM files, or healthcare IT systems.
Albumentations applies image transformations to training data, supporting classification, segmentation, object detection, and pose estimation with a unified API for images, masks, bounding boxes, and keypoints.
Install it if you need a unified, production-grade augmentation API for computer vision tasks.
CoolProp provides thermodynamic and transport property calculations for fluids and fluid mixtures, offering open-source functionality similar to REFPROP.
Provides quaternion representation, manipulation, and rotation operations for 3D geometry and animation, with support for smooth interpolation between orientations.
See also vesin-torch · OpenMM · warp-lang · cudensitymat-cu13 · cuvs-cu12 · vesin · nvidia-cublas-cu11 · nvidia-cusolver-cu11 · newton · custatevec-cu13