PyPI version License GitHub commit activity Downloads codecov GitHub - CI

NVIDIA Warp

Documentation | Changelog

Warp is a Python framework for GPU-accelerated simulation, robotics, and machine learning. Warp takes regular Python functions and JIT compiles them to efficient kernel code that can run on the CPU or GPU.

Warp comes with a rich set of primitives for physics simulation, robotics, geometry processing, and more. Warp kernels are differentiable and can be used as part of machine-learning pipelines with frameworks such as PyTorch, JAX and Paddle.

A selection of physical simulations computed with Warp

Quick Start

Simulate one million particles under gravitational attraction, in 20 lines:

import warp as wp
import numpy as np

num_particles = 1_000_000
dt = 0.01

@wp.kernel
def gravity_step(pos: wp.array[wp.vec3], vel: wp.array[wp.vec3]):
    i = wp.tid()
    position = pos[i]
    dist_sq = wp.length_sq(position) + 0.01  # softened distance
    acc = -1000.0 / dist_sq * wp.normalize(position)  # gravitational pull toward origin
    vel[i] = vel[i] + acc * dt
    pos[i] = pos[i] + vel[i] * dt

rng = np.random.default_rng(42)
positions = wp.array(rng.normal(size=(num_particles, 3)), dtype=wp.vec3)
velocities = wp.array(rng.normal(size=(num_particles, 3)), dtype=wp.vec3)

for _ in range(100):
    wp.launch(gravity_step, dim=num_particles, inputs=[positions, velocities])

print(positions.numpy())

Installing

Python version 3.10 or newer is required. Warp can run on x86-64 and ARMv8 CPUs on Windows and Linux, and on Apple Silicon (ARMv8) on macOS. GPU support requires a CUDA-capable NVIDIA GPU and driver (minimum GeForce GTX 9xx).

The easiest way to install Warp is from PyPI:

pip install warp-lang

You can also use pip install warp-lang[examples] to install additional dependencies for running examples and USD-related features.

For nightly builds, conda, CUDA 13 builds, building from source, and CUDA driver requirements, see the Installation Guide.

Tutorial Notebooks

The NVIDIA Accelerated Computing Hub also hosts Warp tutorial notebooks that can be opened in Colab:

Notebook Colab Link
01. Introduction to NVIDIA Warp Open In Colab
02. Ising Model Open In Colab
03. 2D Navier-Stokes Solver Open In Colab
04. Differentiable Simulations in Warp Open In Colab

Running Examples

The warp/examples directory contains examples covering physics simulation, geometry processing, optimization, and tile-based GPU programming. Before running examples, install the optional example dependencies using:

pip install warp-lang[examples]

On Linux aarch64 systems (e.g., NVIDIA DGX Spark), the [examples] extra automatically installs usd-exchange instead of usd-core as a drop-in replacement, since usd-core wheels are not available for that platform.

Examples can be run from the command-line as follows:

python -m warp.examples.<example_subdir>.<example>

Most examples can be run on either the CPU or a CUDA-capable device, but a handful require a CUDA-capable device. These are marked at the top of the example script. Some examples generate USD files containing time-sampled animations in the current working directory. These can be viewed in Pixar's UsdView, Blender, or any USD-compatible viewer.

To browse the example source code, you can open the directory where the files are located like this:

python -m warp.examples.browse

warp/examples/core

dem fluid graph capture marching cubes
mesh nvdb raycast raymarch
sample mesh sph torch wave
2-D incompressible turbulence in a periodic box

warp/examples/fem

diffusion 3d mixed elasticity apic fluid streamlines
distortion energy taylor green kelvin helmholtz magnetostatics
adaptive grid nonconforming contact darcy level-set optimization elastic shape optimization

warp/examples/optim

diffray fluid checkpoint particle repulsion navier-stokes perturbation

warp/examples/tile

mlp nbody mcgp

Learn More

Please see the following resources for additional background on Warp:

Support

See the FAQ for common questions.

Problems, questions, and feature requests can be opened on GitHub Issues.

For inquiries not suited for GitHub Issues, please email warp-python@nvidia.com.

Contributing

Contributions and pull requests from the community are welcome. Please see the Contribution Guide for more information on contributing to the development of Warp.

License

Warp is provided under the Apache License, Version 2.0. Please see LICENSE.md for full license text.

This project will download and install additional third-party open source software projects. Review the license terms of these open source projects before use.

Building from Source

When building Warp from source using the build_lib.py script, the build process automatically downloads NVIDIA libmathdx. Pre-built Warp packages (e.g., from PyPI) already include libmathdx statically linked into the library binaries. In both cases, libmathdx is governed by the NVIDIA Software License Agreement.

NOTICE AND DISCLAIMER: This software automatically retrieves, accesses or interacts with external materials. Those retrieved materials are not distributed with this software and are governed solely by separate terms, conditions and licenses. You are solely responsible for finding, reviewing and complying with all applicable terms, conditions, and licenses, and for verifying the security, integrity and suitability of any retrieved materials for your specific use case. This software is provided "AS IS", without warranty of any kind. The author makes no representations or warranties regarding any retrieved materials, and assumes no liability for any losses, damages, liabilities or legal consequences from your use or inability to use this software or any retrieved materials. Use this software and the retrieved materials at your own risk.

Publications & Citation

Research Using Warp

Our PUBLICATIONS.md file lists academic and research publications that leverage the capabilities of Warp. We encourage you to add your own published work using Warp to this list.

Citing Warp

If you use Warp in your research, please use the "Cite this repository" button on the GitHub repository page or refer to the CITATION.cff file for citation information.

Project links

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (1.17.0)

Owner

PyPI data
Data sourced directly from PyPI's database.

2 maintainers

PyPI data
Data sourced directly from PyPI's database.
Avatar for eshi from gravatar.com eshi Avatar for mmacklin from gravatar.com mmacklin

Credits

Author: NVIDIA Corporation

License

Apache-2.0

Requires

Python >=3.10

Provides Extra

docs benchmark examples torch-cu12 torch-cu13 dev

Classifiers

Report project as malware

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

warp_lang-1.16.0-py3-none-win_amd64.whl (146.4 MB view details)

Uploaded Python 3Windows x86-64

warp_lang-1.16.0-py3-none-manylinux_2_34_aarch64.whl (167.6 MB view details)

Uploaded Python 3manylinux: glibc 2.34+ ARM64

warp_lang-1.16.0-py3-none-manylinux_2_28_x86_64.whl (163.1 MB view details)

Uploaded Python 3manylinux: glibc 2.28+ x86-64

warp_lang-1.16.0-py3-none-macosx_11_0_arm64.whl (25.2 MB view details)

Uploaded Python 3macOS 11.0+ ARM64

File details

Details for the file warp_lang-1.16.0-py3-none-win_amd64.whl.

File metadata

  • Download URL: warp_lang-1.16.0-py3-none-win_amd64.whl
  • Upload date:
  • Size: 146.4 MB
  • Tags: Python 3, Windows x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.6

File hashes

Hashes for warp_lang-1.16.0-py3-none-win_amd64.whl
Algorithm Hash digest
SHA256 8690c9096e0a271985339d4aa4b37675e7d62852620ca861ac032dc85b124542
MD5 10304caed15e66955ff56c9bce974557
BLAKE2b-256 7e17c98fab81156f0d25f17200f52f69210b94a9abba101066736df8cc493522

See more details on using hashes here.

File details

Details for the file warp_lang-1.16.0-py3-none-manylinux_2_34_aarch64.whl.

File metadata

File hashes

Hashes for warp_lang-1.16.0-py3-none-manylinux_2_34_aarch64.whl
Algorithm Hash digest
SHA256 d4715171bd6821436b82293d774c9a082ace08215c189572b4348d1e48fb8ee8
MD5 4ba9adfc6e242b0ec28eb1c940315ec0
BLAKE2b-256 db960c2b070b3101c669955cafd36262548b2e8b00dfdeda3aeb7afbc0f9d65c

See more details on using hashes here.

File details

Details for the file warp_lang-1.16.0-py3-none-manylinux_2_28_x86_64.whl.

File metadata

File hashes

Hashes for warp_lang-1.16.0-py3-none-manylinux_2_28_x86_64.whl
Algorithm Hash digest
SHA256 96449fc1e3b354185e2f09434fb794b5953ab2e8673b104d7e7f48d5d418bb35
MD5 397d85e81ed8208a5a85205d3dc2d984
BLAKE2b-256 1d3d47feef2eee4739437e19d6949ea231ca4181c9eacbaa39144761e6130d94

See more details on using hashes here.

File details

Details for the file warp_lang-1.16.0-py3-none-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for warp_lang-1.16.0-py3-none-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 19a7590c4484e8250ab19165311d2a1774138663b361c41e8be2865c7515c4ae
MD5 5c40de502701d9fd26dec317070a7b7b
BLAKE2b-256 738a8be711f3b5431a6e0bdb97117d681b03cf2bb95d6fcf8869861900898d88

See more details on using hashes here.

Release history Release notifications | RSS feed

1.17.0

4 files

This release

1.16.0 This release

4 files

1.15.0

4 files

1.14.0

4 files

1.13.0

4 files

1.12.1

4 files

1.12.0

4 files

1.11.1

4 files

1.11.0

4 files

1.10.1

4 files

1.10.0

4 files

1.9.1

4 files

1.9.0

4 files

1.8.1

4 files

1.8.0

4 files

1.7.2

4 files

1.7.1

4 files

1.7.0

4 files

1.6.2

4 files

1.6.1

4 files

1.6.0

4 files

1.5.1

4 files

1.5.0

4 files

1.4.2

4 files

1.4.1

4 files

1.4.0

4 files

1.3.3

4 files

1.3.2

4 files

1.3.1

4 files

1.3.0

4 files

1.2.2

4 files

1.2.1

4 files

1.2.0

4 files

1.1.0

4 files

1.0.2

4 files

1.0.1

4 files

1.0.0

4 files

0.15.1

4 files

0.15.0

4 files

0.13.0

4 files

0.11.0

4 files

0.10.1

3 files

0.9.0

3 files

0.8.0

2 files

0.7.2

2 files

0.6.2

2 files

0.6.1

2 files

0.5.0

2 files

0.4.3

2 files

0.3.1

2 files

0.2.3

2 files

0.2.0

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page