warp-lang 1.7.0__py3-none-manylinux_2_34_aarch64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of warp-lang might be problematic. Click here for more details.
- warp/__init__.py +139 -0
- warp/__init__.pyi +1 -0
- warp/autograd.py +1142 -0
- warp/bin/warp-clang.so +0 -0
- warp/bin/warp.so +0 -0
- warp/build.py +557 -0
- warp/build_dll.py +405 -0
- warp/builtins.py +6855 -0
- warp/codegen.py +3969 -0
- warp/config.py +158 -0
- warp/constants.py +57 -0
- warp/context.py +6812 -0
- warp/dlpack.py +462 -0
- warp/examples/__init__.py +24 -0
- warp/examples/assets/bear.usd +0 -0
- warp/examples/assets/bunny.usd +0 -0
- warp/examples/assets/cartpole.urdf +110 -0
- warp/examples/assets/crazyflie.usd +0 -0
- warp/examples/assets/cube.usd +0 -0
- warp/examples/assets/nonuniform.usd +0 -0
- warp/examples/assets/nv_ant.xml +92 -0
- warp/examples/assets/nv_humanoid.xml +183 -0
- warp/examples/assets/nvidia_logo.png +0 -0
- warp/examples/assets/pixel.jpg +0 -0
- warp/examples/assets/quadruped.urdf +268 -0
- warp/examples/assets/rocks.nvdb +0 -0
- warp/examples/assets/rocks.usd +0 -0
- warp/examples/assets/sphere.usd +0 -0
- warp/examples/assets/square_cloth.usd +0 -0
- warp/examples/benchmarks/benchmark_api.py +389 -0
- warp/examples/benchmarks/benchmark_cloth.py +296 -0
- warp/examples/benchmarks/benchmark_cloth_cupy.py +96 -0
- warp/examples/benchmarks/benchmark_cloth_jax.py +105 -0
- warp/examples/benchmarks/benchmark_cloth_numba.py +161 -0
- warp/examples/benchmarks/benchmark_cloth_numpy.py +85 -0
- warp/examples/benchmarks/benchmark_cloth_paddle.py +94 -0
- warp/examples/benchmarks/benchmark_cloth_pytorch.py +94 -0
- warp/examples/benchmarks/benchmark_cloth_taichi.py +120 -0
- warp/examples/benchmarks/benchmark_cloth_warp.py +153 -0
- warp/examples/benchmarks/benchmark_gemm.py +164 -0
- warp/examples/benchmarks/benchmark_interop_paddle.py +166 -0
- warp/examples/benchmarks/benchmark_interop_torch.py +166 -0
- warp/examples/benchmarks/benchmark_launches.py +301 -0
- warp/examples/benchmarks/benchmark_tile_load_store.py +103 -0
- warp/examples/browse.py +37 -0
- warp/examples/core/example_cupy.py +86 -0
- warp/examples/core/example_dem.py +241 -0
- warp/examples/core/example_fluid.py +299 -0
- warp/examples/core/example_graph_capture.py +150 -0
- warp/examples/core/example_marching_cubes.py +194 -0
- warp/examples/core/example_mesh.py +180 -0
- warp/examples/core/example_mesh_intersect.py +211 -0
- warp/examples/core/example_nvdb.py +182 -0
- warp/examples/core/example_raycast.py +111 -0
- warp/examples/core/example_raymarch.py +205 -0
- warp/examples/core/example_render_opengl.py +193 -0
- warp/examples/core/example_sample_mesh.py +300 -0
- warp/examples/core/example_sph.py +411 -0
- warp/examples/core/example_torch.py +211 -0
- warp/examples/core/example_wave.py +269 -0
- warp/examples/fem/example_adaptive_grid.py +286 -0
- warp/examples/fem/example_apic_fluid.py +423 -0
- warp/examples/fem/example_burgers.py +261 -0
- warp/examples/fem/example_convection_diffusion.py +178 -0
- warp/examples/fem/example_convection_diffusion_dg.py +204 -0
- warp/examples/fem/example_deformed_geometry.py +172 -0
- warp/examples/fem/example_diffusion.py +196 -0
- warp/examples/fem/example_diffusion_3d.py +225 -0
- warp/examples/fem/example_diffusion_mgpu.py +220 -0
- warp/examples/fem/example_distortion_energy.py +228 -0
- warp/examples/fem/example_magnetostatics.py +240 -0
- warp/examples/fem/example_mixed_elasticity.py +291 -0
- warp/examples/fem/example_navier_stokes.py +261 -0
- warp/examples/fem/example_nonconforming_contact.py +298 -0
- warp/examples/fem/example_stokes.py +213 -0
- warp/examples/fem/example_stokes_transfer.py +262 -0
- warp/examples/fem/example_streamlines.py +352 -0
- warp/examples/fem/utils.py +1000 -0
- warp/examples/interop/example_jax_callable.py +116 -0
- warp/examples/interop/example_jax_ffi_callback.py +132 -0
- warp/examples/interop/example_jax_kernel.py +205 -0
- warp/examples/optim/example_bounce.py +266 -0
- warp/examples/optim/example_cloth_throw.py +228 -0
- warp/examples/optim/example_diffray.py +561 -0
- warp/examples/optim/example_drone.py +870 -0
- warp/examples/optim/example_fluid_checkpoint.py +497 -0
- warp/examples/optim/example_inverse_kinematics.py +182 -0
- warp/examples/optim/example_inverse_kinematics_torch.py +191 -0
- warp/examples/optim/example_softbody_properties.py +400 -0
- warp/examples/optim/example_spring_cage.py +245 -0
- warp/examples/optim/example_trajectory.py +227 -0
- warp/examples/sim/example_cartpole.py +143 -0
- warp/examples/sim/example_cloth.py +225 -0
- warp/examples/sim/example_cloth_self_contact.py +322 -0
- warp/examples/sim/example_granular.py +130 -0
- warp/examples/sim/example_granular_collision_sdf.py +202 -0
- warp/examples/sim/example_jacobian_ik.py +244 -0
- warp/examples/sim/example_particle_chain.py +124 -0
- warp/examples/sim/example_quadruped.py +203 -0
- warp/examples/sim/example_rigid_chain.py +203 -0
- warp/examples/sim/example_rigid_contact.py +195 -0
- warp/examples/sim/example_rigid_force.py +133 -0
- warp/examples/sim/example_rigid_gyroscopic.py +115 -0
- warp/examples/sim/example_rigid_soft_contact.py +140 -0
- warp/examples/sim/example_soft_body.py +196 -0
- warp/examples/tile/example_tile_cholesky.py +87 -0
- warp/examples/tile/example_tile_convolution.py +66 -0
- warp/examples/tile/example_tile_fft.py +55 -0
- warp/examples/tile/example_tile_filtering.py +113 -0
- warp/examples/tile/example_tile_matmul.py +85 -0
- warp/examples/tile/example_tile_mlp.py +383 -0
- warp/examples/tile/example_tile_nbody.py +199 -0
- warp/examples/tile/example_tile_walker.py +327 -0
- warp/fabric.py +355 -0
- warp/fem/__init__.py +106 -0
- warp/fem/adaptivity.py +508 -0
- warp/fem/cache.py +572 -0
- warp/fem/dirichlet.py +202 -0
- warp/fem/domain.py +411 -0
- warp/fem/field/__init__.py +125 -0
- warp/fem/field/field.py +619 -0
- warp/fem/field/nodal_field.py +326 -0
- warp/fem/field/restriction.py +37 -0
- warp/fem/field/virtual.py +848 -0
- warp/fem/geometry/__init__.py +32 -0
- warp/fem/geometry/adaptive_nanogrid.py +857 -0
- warp/fem/geometry/closest_point.py +84 -0
- warp/fem/geometry/deformed_geometry.py +221 -0
- warp/fem/geometry/element.py +776 -0
- warp/fem/geometry/geometry.py +362 -0
- warp/fem/geometry/grid_2d.py +392 -0
- warp/fem/geometry/grid_3d.py +452 -0
- warp/fem/geometry/hexmesh.py +911 -0
- warp/fem/geometry/nanogrid.py +571 -0
- warp/fem/geometry/partition.py +389 -0
- warp/fem/geometry/quadmesh.py +663 -0
- warp/fem/geometry/tetmesh.py +855 -0
- warp/fem/geometry/trimesh.py +806 -0
- warp/fem/integrate.py +2335 -0
- warp/fem/linalg.py +419 -0
- warp/fem/operator.py +293 -0
- warp/fem/polynomial.py +229 -0
- warp/fem/quadrature/__init__.py +17 -0
- warp/fem/quadrature/pic_quadrature.py +299 -0
- warp/fem/quadrature/quadrature.py +591 -0
- warp/fem/space/__init__.py +228 -0
- warp/fem/space/basis_function_space.py +468 -0
- warp/fem/space/basis_space.py +667 -0
- warp/fem/space/dof_mapper.py +251 -0
- warp/fem/space/function_space.py +309 -0
- warp/fem/space/grid_2d_function_space.py +177 -0
- warp/fem/space/grid_3d_function_space.py +227 -0
- warp/fem/space/hexmesh_function_space.py +257 -0
- warp/fem/space/nanogrid_function_space.py +201 -0
- warp/fem/space/partition.py +367 -0
- warp/fem/space/quadmesh_function_space.py +223 -0
- warp/fem/space/restriction.py +179 -0
- warp/fem/space/shape/__init__.py +143 -0
- warp/fem/space/shape/cube_shape_function.py +1105 -0
- warp/fem/space/shape/shape_function.py +133 -0
- warp/fem/space/shape/square_shape_function.py +926 -0
- warp/fem/space/shape/tet_shape_function.py +834 -0
- warp/fem/space/shape/triangle_shape_function.py +672 -0
- warp/fem/space/tetmesh_function_space.py +271 -0
- warp/fem/space/topology.py +424 -0
- warp/fem/space/trimesh_function_space.py +194 -0
- warp/fem/types.py +99 -0
- warp/fem/utils.py +420 -0
- warp/jax.py +187 -0
- warp/jax_experimental/__init__.py +16 -0
- warp/jax_experimental/custom_call.py +351 -0
- warp/jax_experimental/ffi.py +698 -0
- warp/jax_experimental/xla_ffi.py +602 -0
- warp/math.py +244 -0
- warp/native/array.h +1145 -0
- warp/native/builtin.h +1800 -0
- warp/native/bvh.cpp +492 -0
- warp/native/bvh.cu +791 -0
- warp/native/bvh.h +554 -0
- warp/native/clang/clang.cpp +536 -0
- warp/native/coloring.cpp +613 -0
- warp/native/crt.cpp +51 -0
- warp/native/crt.h +362 -0
- warp/native/cuda_crt.h +1058 -0
- warp/native/cuda_util.cpp +646 -0
- warp/native/cuda_util.h +307 -0
- warp/native/error.cpp +77 -0
- warp/native/error.h +36 -0
- warp/native/exports.h +1878 -0
- warp/native/fabric.h +245 -0
- warp/native/hashgrid.cpp +311 -0
- warp/native/hashgrid.cu +87 -0
- warp/native/hashgrid.h +240 -0
- warp/native/initializer_array.h +41 -0
- warp/native/intersect.h +1230 -0
- warp/native/intersect_adj.h +375 -0
- warp/native/intersect_tri.h +339 -0
- warp/native/marching.cpp +19 -0
- warp/native/marching.cu +514 -0
- warp/native/marching.h +19 -0
- warp/native/mat.h +2220 -0
- warp/native/mathdx.cpp +87 -0
- warp/native/matnn.h +343 -0
- warp/native/mesh.cpp +266 -0
- warp/native/mesh.cu +404 -0
- warp/native/mesh.h +1980 -0
- warp/native/nanovdb/GridHandle.h +366 -0
- warp/native/nanovdb/HostBuffer.h +590 -0
- warp/native/nanovdb/NanoVDB.h +6624 -0
- warp/native/nanovdb/PNanoVDB.h +3390 -0
- warp/native/noise.h +859 -0
- warp/native/quat.h +1371 -0
- warp/native/rand.h +342 -0
- warp/native/range.h +139 -0
- warp/native/reduce.cpp +174 -0
- warp/native/reduce.cu +364 -0
- warp/native/runlength_encode.cpp +79 -0
- warp/native/runlength_encode.cu +61 -0
- warp/native/scan.cpp +47 -0
- warp/native/scan.cu +53 -0
- warp/native/scan.h +23 -0
- warp/native/solid_angle.h +466 -0
- warp/native/sort.cpp +251 -0
- warp/native/sort.cu +277 -0
- warp/native/sort.h +33 -0
- warp/native/sparse.cpp +378 -0
- warp/native/sparse.cu +524 -0
- warp/native/spatial.h +657 -0
- warp/native/svd.h +702 -0
- warp/native/temp_buffer.h +46 -0
- warp/native/tile.h +2584 -0
- warp/native/tile_reduce.h +264 -0
- warp/native/vec.h +1426 -0
- warp/native/volume.cpp +501 -0
- warp/native/volume.cu +67 -0
- warp/native/volume.h +969 -0
- warp/native/volume_builder.cu +477 -0
- warp/native/volume_builder.h +52 -0
- warp/native/volume_impl.h +70 -0
- warp/native/warp.cpp +1082 -0
- warp/native/warp.cu +3636 -0
- warp/native/warp.h +381 -0
- warp/optim/__init__.py +17 -0
- warp/optim/adam.py +163 -0
- warp/optim/linear.py +1137 -0
- warp/optim/sgd.py +112 -0
- warp/paddle.py +407 -0
- warp/render/__init__.py +18 -0
- warp/render/render_opengl.py +3518 -0
- warp/render/render_usd.py +784 -0
- warp/render/utils.py +160 -0
- warp/sim/__init__.py +65 -0
- warp/sim/articulation.py +793 -0
- warp/sim/collide.py +2395 -0
- warp/sim/graph_coloring.py +300 -0
- warp/sim/import_mjcf.py +790 -0
- warp/sim/import_snu.py +227 -0
- warp/sim/import_urdf.py +579 -0
- warp/sim/import_usd.py +894 -0
- warp/sim/inertia.py +324 -0
- warp/sim/integrator.py +242 -0
- warp/sim/integrator_euler.py +1997 -0
- warp/sim/integrator_featherstone.py +2101 -0
- warp/sim/integrator_vbd.py +2048 -0
- warp/sim/integrator_xpbd.py +3292 -0
- warp/sim/model.py +4791 -0
- warp/sim/particles.py +121 -0
- warp/sim/render.py +427 -0
- warp/sim/utils.py +428 -0
- warp/sparse.py +2057 -0
- warp/stubs.py +3333 -0
- warp/tape.py +1203 -0
- warp/tests/__init__.py +1 -0
- warp/tests/__main__.py +4 -0
- warp/tests/assets/curlnoise_golden.npy +0 -0
- warp/tests/assets/mlp_golden.npy +0 -0
- warp/tests/assets/pixel.npy +0 -0
- warp/tests/assets/pnoise_golden.npy +0 -0
- warp/tests/assets/spiky.usd +0 -0
- warp/tests/assets/test_grid.nvdb +0 -0
- warp/tests/assets/test_index_grid.nvdb +0 -0
- warp/tests/assets/test_int32_grid.nvdb +0 -0
- warp/tests/assets/test_vec_grid.nvdb +0 -0
- warp/tests/assets/torus.nvdb +0 -0
- warp/tests/assets/torus.usda +105 -0
- warp/tests/aux_test_class_kernel.py +34 -0
- warp/tests/aux_test_compile_consts_dummy.py +18 -0
- warp/tests/aux_test_conditional_unequal_types_kernels.py +29 -0
- warp/tests/aux_test_dependent.py +29 -0
- warp/tests/aux_test_grad_customs.py +29 -0
- warp/tests/aux_test_instancing_gc.py +26 -0
- warp/tests/aux_test_module_unload.py +23 -0
- warp/tests/aux_test_name_clash1.py +40 -0
- warp/tests/aux_test_name_clash2.py +40 -0
- warp/tests/aux_test_reference.py +9 -0
- warp/tests/aux_test_reference_reference.py +8 -0
- warp/tests/aux_test_square.py +16 -0
- warp/tests/aux_test_unresolved_func.py +22 -0
- warp/tests/aux_test_unresolved_symbol.py +22 -0
- warp/tests/cuda/__init__.py +0 -0
- warp/tests/cuda/test_async.py +676 -0
- warp/tests/cuda/test_ipc.py +124 -0
- warp/tests/cuda/test_mempool.py +233 -0
- warp/tests/cuda/test_multigpu.py +169 -0
- warp/tests/cuda/test_peer.py +139 -0
- warp/tests/cuda/test_pinned.py +84 -0
- warp/tests/cuda/test_streams.py +634 -0
- warp/tests/geometry/__init__.py +0 -0
- warp/tests/geometry/test_bvh.py +200 -0
- warp/tests/geometry/test_hash_grid.py +221 -0
- warp/tests/geometry/test_marching_cubes.py +74 -0
- warp/tests/geometry/test_mesh.py +316 -0
- warp/tests/geometry/test_mesh_query_aabb.py +399 -0
- warp/tests/geometry/test_mesh_query_point.py +932 -0
- warp/tests/geometry/test_mesh_query_ray.py +311 -0
- warp/tests/geometry/test_volume.py +1103 -0
- warp/tests/geometry/test_volume_write.py +346 -0
- warp/tests/interop/__init__.py +0 -0
- warp/tests/interop/test_dlpack.py +729 -0
- warp/tests/interop/test_jax.py +371 -0
- warp/tests/interop/test_paddle.py +800 -0
- warp/tests/interop/test_torch.py +1001 -0
- warp/tests/run_coverage_serial.py +39 -0
- warp/tests/sim/__init__.py +0 -0
- warp/tests/sim/disabled_kinematics.py +244 -0
- warp/tests/sim/flaky_test_sim_grad.py +290 -0
- warp/tests/sim/test_collision.py +604 -0
- warp/tests/sim/test_coloring.py +258 -0
- warp/tests/sim/test_model.py +224 -0
- warp/tests/sim/test_sim_grad_bounce_linear.py +212 -0
- warp/tests/sim/test_sim_kinematics.py +98 -0
- warp/tests/sim/test_vbd.py +597 -0
- warp/tests/test_adam.py +163 -0
- warp/tests/test_arithmetic.py +1096 -0
- warp/tests/test_array.py +2972 -0
- warp/tests/test_array_reduce.py +156 -0
- warp/tests/test_assert.py +250 -0
- warp/tests/test_atomic.py +153 -0
- warp/tests/test_bool.py +220 -0
- warp/tests/test_builtins_resolution.py +1298 -0
- warp/tests/test_closest_point_edge_edge.py +327 -0
- warp/tests/test_codegen.py +810 -0
- warp/tests/test_codegen_instancing.py +1495 -0
- warp/tests/test_compile_consts.py +215 -0
- warp/tests/test_conditional.py +252 -0
- warp/tests/test_context.py +42 -0
- warp/tests/test_copy.py +238 -0
- warp/tests/test_ctypes.py +638 -0
- warp/tests/test_dense.py +73 -0
- warp/tests/test_devices.py +97 -0
- warp/tests/test_examples.py +482 -0
- warp/tests/test_fabricarray.py +996 -0
- warp/tests/test_fast_math.py +74 -0
- warp/tests/test_fem.py +2003 -0
- warp/tests/test_fp16.py +136 -0
- warp/tests/test_func.py +454 -0
- warp/tests/test_future_annotations.py +98 -0
- warp/tests/test_generics.py +656 -0
- warp/tests/test_grad.py +893 -0
- warp/tests/test_grad_customs.py +339 -0
- warp/tests/test_grad_debug.py +341 -0
- warp/tests/test_implicit_init.py +411 -0
- warp/tests/test_import.py +45 -0
- warp/tests/test_indexedarray.py +1140 -0
- warp/tests/test_intersect.py +73 -0
- warp/tests/test_iter.py +76 -0
- warp/tests/test_large.py +177 -0
- warp/tests/test_launch.py +411 -0
- warp/tests/test_lerp.py +151 -0
- warp/tests/test_linear_solvers.py +193 -0
- warp/tests/test_lvalue.py +427 -0
- warp/tests/test_mat.py +2089 -0
- warp/tests/test_mat_lite.py +122 -0
- warp/tests/test_mat_scalar_ops.py +2913 -0
- warp/tests/test_math.py +178 -0
- warp/tests/test_mlp.py +282 -0
- warp/tests/test_module_hashing.py +258 -0
- warp/tests/test_modules_lite.py +44 -0
- warp/tests/test_noise.py +252 -0
- warp/tests/test_operators.py +299 -0
- warp/tests/test_options.py +129 -0
- warp/tests/test_overwrite.py +551 -0
- warp/tests/test_print.py +339 -0
- warp/tests/test_quat.py +2315 -0
- warp/tests/test_rand.py +339 -0
- warp/tests/test_reload.py +302 -0
- warp/tests/test_rounding.py +185 -0
- warp/tests/test_runlength_encode.py +196 -0
- warp/tests/test_scalar_ops.py +105 -0
- warp/tests/test_smoothstep.py +108 -0
- warp/tests/test_snippet.py +318 -0
- warp/tests/test_sparse.py +582 -0
- warp/tests/test_spatial.py +2229 -0
- warp/tests/test_special_values.py +361 -0
- warp/tests/test_static.py +592 -0
- warp/tests/test_struct.py +734 -0
- warp/tests/test_tape.py +204 -0
- warp/tests/test_transient_module.py +93 -0
- warp/tests/test_triangle_closest_point.py +145 -0
- warp/tests/test_types.py +562 -0
- warp/tests/test_utils.py +588 -0
- warp/tests/test_vec.py +1487 -0
- warp/tests/test_vec_lite.py +80 -0
- warp/tests/test_vec_scalar_ops.py +2327 -0
- warp/tests/test_verify_fp.py +100 -0
- warp/tests/tile/__init__.py +0 -0
- warp/tests/tile/test_tile.py +780 -0
- warp/tests/tile/test_tile_load.py +407 -0
- warp/tests/tile/test_tile_mathdx.py +208 -0
- warp/tests/tile/test_tile_mlp.py +402 -0
- warp/tests/tile/test_tile_reduce.py +447 -0
- warp/tests/tile/test_tile_shared_memory.py +247 -0
- warp/tests/tile/test_tile_view.py +173 -0
- warp/tests/unittest_serial.py +47 -0
- warp/tests/unittest_suites.py +427 -0
- warp/tests/unittest_utils.py +468 -0
- warp/tests/walkthrough_debug.py +93 -0
- warp/thirdparty/__init__.py +0 -0
- warp/thirdparty/appdirs.py +598 -0
- warp/thirdparty/dlpack.py +145 -0
- warp/thirdparty/unittest_parallel.py +570 -0
- warp/torch.py +391 -0
- warp/types.py +5230 -0
- warp/utils.py +1137 -0
- warp_lang-1.7.0.dist-info/METADATA +516 -0
- warp_lang-1.7.0.dist-info/RECORD +429 -0
- warp_lang-1.7.0.dist-info/WHEEL +5 -0
- warp_lang-1.7.0.dist-info/licenses/LICENSE.md +202 -0
- warp_lang-1.7.0.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,516 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: warp-lang
|
|
3
|
+
Version: 1.7.0
|
|
4
|
+
Summary: A Python framework for high-performance simulation and graphics programming
|
|
5
|
+
Author-email: NVIDIA Corporation <warp-python@nvidia.com>
|
|
6
|
+
License: Apache-2.0
|
|
7
|
+
Project-URL: GitHub, https://github.com/NVIDIA/warp
|
|
8
|
+
Project-URL: Documentation, https://nvidia.github.io/warp
|
|
9
|
+
Project-URL: Changelog, https://github.com/NVIDIA/warp/blob/main/CHANGELOG.md
|
|
10
|
+
Classifier: Programming Language :: Python :: 3.8
|
|
11
|
+
Classifier: Programming Language :: Python :: 3.9
|
|
12
|
+
Classifier: Programming Language :: Python :: 3.10
|
|
13
|
+
Classifier: Programming Language :: Python :: 3.11
|
|
14
|
+
Classifier: Programming Language :: Python :: 3.12
|
|
15
|
+
Classifier: Programming Language :: Python :: 3.13
|
|
16
|
+
Classifier: Programming Language :: Python :: 3 :: Only
|
|
17
|
+
Classifier: License :: OSI Approved :: Apache Software License
|
|
18
|
+
Classifier: Operating System :: OS Independent
|
|
19
|
+
Requires-Python: >=3.8
|
|
20
|
+
Description-Content-Type: text/markdown
|
|
21
|
+
License-File: LICENSE.md
|
|
22
|
+
Requires-Dist: numpy
|
|
23
|
+
Provides-Extra: dev
|
|
24
|
+
Requires-Dist: pre-commit; extra == "dev"
|
|
25
|
+
Requires-Dist: ruff; extra == "dev"
|
|
26
|
+
Requires-Dist: nvtx; extra == "dev"
|
|
27
|
+
Requires-Dist: nvidia-sphinx-theme; python_version >= "3.9" and extra == "dev"
|
|
28
|
+
Requires-Dist: sphinx-copybutton; extra == "dev"
|
|
29
|
+
Requires-Dist: coverage[toml]; extra == "dev"
|
|
30
|
+
Provides-Extra: extras
|
|
31
|
+
Requires-Dist: usd-core; extra == "extras"
|
|
32
|
+
Requires-Dist: matplotlib; extra == "extras"
|
|
33
|
+
Requires-Dist: pillow; extra == "extras"
|
|
34
|
+
Requires-Dist: pyglet; extra == "extras"
|
|
35
|
+
Dynamic: license-file
|
|
36
|
+
|
|
37
|
+
[](https://badge.fury.io/py/warp-lang)
|
|
38
|
+
[](https://opensource.org/licenses/Apache-2.0)
|
|
39
|
+

|
|
40
|
+
[](https://pepy.tech/project/warp-lang)
|
|
41
|
+
[](https://codecov.io/github/NVIDIA/warp)
|
|
42
|
+

|
|
43
|
+
[](https://discord.com/invite/nvidiaomniverse)
|
|
44
|
+
|
|
45
|
+
# NVIDIA Warp
|
|
46
|
+
|
|
47
|
+
Warp is a Python framework for writing high-performance simulation and graphics code. Warp takes
|
|
48
|
+
regular Python functions and JIT compiles them to efficient kernel code that can run on the CPU or GPU.
|
|
49
|
+
|
|
50
|
+
Warp is designed for [spatial computing](https://en.wikipedia.org/wiki/Spatial_computing)
|
|
51
|
+
and comes with a rich set of primitives that make it easy to write
|
|
52
|
+
programs for physics simulation, perception, robotics, and geometry processing. In addition, Warp kernels
|
|
53
|
+
are differentiable and can be used as part of machine-learning pipelines with frameworks such as PyTorch, JAX and Paddle.
|
|
54
|
+
|
|
55
|
+
Please refer to the project [Documentation](https://nvidia.github.io/warp/) for API and language reference and [CHANGELOG.md](./CHANGELOG.md) for release history.
|
|
56
|
+
|
|
57
|
+
<div align="center">
|
|
58
|
+
<img src="https://github.com/NVIDIA/warp/raw/main/docs/img/header.jpg">
|
|
59
|
+
<p><i>A selection of physical simulations computed with Warp</i></p>
|
|
60
|
+
</div>
|
|
61
|
+
|
|
62
|
+
## Installing
|
|
63
|
+
|
|
64
|
+
Python version 3.9 or newer is recommended. Warp can run on x86-64 and ARMv8 CPUs on Windows, Linux, and macOS.
|
|
65
|
+
GPU support requires a CUDA-capable NVIDIA GPU and driver (minimum GeForce GTX 9xx).
|
|
66
|
+
|
|
67
|
+
The easiest way to install Warp is from [PyPI](https://pypi.org/project/warp-lang/):
|
|
68
|
+
|
|
69
|
+
```text
|
|
70
|
+
pip install warp-lang
|
|
71
|
+
```
|
|
72
|
+
|
|
73
|
+
You can also use `pip install warp-lang[extras]` to install additional dependencies for running examples and USD-related features.
|
|
74
|
+
|
|
75
|
+
The binaries hosted on PyPI are currently built with the CUDA 12 runtime and therefore
|
|
76
|
+
require a minimum version of the CUDA driver of 525.60.13 (Linux x86-64) or 528.33 (Windows x86-64).
|
|
77
|
+
|
|
78
|
+
If you require GPU support on a system with an older CUDA driver, you can build Warp from source or
|
|
79
|
+
install wheels built with the CUDA 11.8 runtime from the [GitHub Releases](https://github.com/NVIDIA/warp/releases) page.
|
|
80
|
+
Copy the URL of the appropriate wheel file (`warp-lang-{ver}+cu12-py3-none-{platform}.whl`) and pass it to
|
|
81
|
+
the `pip install` command, e.g.
|
|
82
|
+
|
|
83
|
+
| Platform | Install Command |
|
|
84
|
+
| --------------- | ----------------------------------------------------------------------------------------------------------------------------- |
|
|
85
|
+
| Linux aarch64 | `pip install https://github.com/NVIDIA/warp/releases/download/v1.7.0/warp_lang-1.7.0+cu11-py3-none-manylinux2014_aarch64.whl` |
|
|
86
|
+
| Linux x86-64 | `pip install https://github.com/NVIDIA/warp/releases/download/v1.7.0/warp_lang-1.7.0+cu11-py3-none-manylinux2014_x86_64.whl` |
|
|
87
|
+
| Windows x86-64 | `pip install https://github.com/NVIDIA/warp/releases/download/v1.7.0/warp_lang-1.7.0+cu11-py3-none-win_amd64.whl` |
|
|
88
|
+
|
|
89
|
+
The `--force-reinstall` option may need to be used to overwrite a previous installation.
|
|
90
|
+
|
|
91
|
+
### Nightly Builds
|
|
92
|
+
|
|
93
|
+
Nightly builds of Warp from the `main` branch are available on the [NVIDIA Package Index](https://pypi.nvidia.com/warp-lang/).
|
|
94
|
+
|
|
95
|
+
To install the latest nightly build, use the following command:
|
|
96
|
+
|
|
97
|
+
```text
|
|
98
|
+
pip install -U --pre warp-lang --extra-index-url=https://pypi.nvidia.com/
|
|
99
|
+
```
|
|
100
|
+
|
|
101
|
+
Note that the nightly builds are built with the CUDA 12 runtime and are not published for macOS.
|
|
102
|
+
|
|
103
|
+
If you plan to install nightly builds regularly, you can simplify future installations by adding NVIDIA's package
|
|
104
|
+
repository as an extra index via the `PIP_EXTRA_INDEX_URL` environment variable. For example:
|
|
105
|
+
|
|
106
|
+
```text
|
|
107
|
+
export PIP_EXTRA_INDEX_URL="https://pypi.nvidia.com"
|
|
108
|
+
```
|
|
109
|
+
|
|
110
|
+
This ensures the index is automatically used for `pip` commands, avoiding the need to specify it explicitly.
|
|
111
|
+
|
|
112
|
+
### CUDA Requirements
|
|
113
|
+
|
|
114
|
+
* Warp packages built with CUDA Toolkit 11.x require NVIDIA driver 470 or newer.
|
|
115
|
+
* Warp packages built with CUDA Toolkit 12.x require NVIDIA driver 525 or newer.
|
|
116
|
+
|
|
117
|
+
This applies to pre-built packages distributed on PyPI and GitHub and also when building Warp from source.
|
|
118
|
+
|
|
119
|
+
Note that building Warp with the `--quick` flag changes the driver requirements. The quick build skips CUDA backward compatibility, so the minimum required driver is determined by the CUDA Toolkit version. Refer to the [latest CUDA Toolkit release notes](https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html) to find the minimum required driver for different CUDA Toolkit versions (e.g., [this table from CUDA Toolkit 12.6](https://docs.nvidia.com/cuda/archive/12.6.0/cuda-toolkit-release-notes/index.html#id5)).
|
|
120
|
+
|
|
121
|
+
Warp checks the installed driver during initialization and will report a warning if the driver is not suitable, e.g.:
|
|
122
|
+
|
|
123
|
+
```text
|
|
124
|
+
Warp UserWarning:
|
|
125
|
+
Insufficient CUDA driver version.
|
|
126
|
+
The minimum required CUDA driver version is 12.0, but the installed CUDA driver version is 11.8.
|
|
127
|
+
Visit https://github.com/NVIDIA/warp/blob/main/README.md#installing for guidance.
|
|
128
|
+
```
|
|
129
|
+
|
|
130
|
+
This will make CUDA devices unavailable, but the CPU can still be used.
|
|
131
|
+
|
|
132
|
+
To remedy the situation there are a few options:
|
|
133
|
+
|
|
134
|
+
* Update the driver.
|
|
135
|
+
* Install a compatible pre-built Warp package.
|
|
136
|
+
* Build Warp from source using a CUDA Toolkit that's compatible with the installed driver.
|
|
137
|
+
|
|
138
|
+
## Getting Started
|
|
139
|
+
|
|
140
|
+
An example first program that computes the lengths of random 3D vectors is given below:
|
|
141
|
+
|
|
142
|
+
```python
|
|
143
|
+
import warp as wp
|
|
144
|
+
import numpy as np
|
|
145
|
+
|
|
146
|
+
num_points = 1024
|
|
147
|
+
|
|
148
|
+
@wp.kernel
|
|
149
|
+
def length(points: wp.array(dtype=wp.vec3),
|
|
150
|
+
lengths: wp.array(dtype=float)):
|
|
151
|
+
|
|
152
|
+
# thread index
|
|
153
|
+
tid = wp.tid()
|
|
154
|
+
|
|
155
|
+
# compute distance of each point from origin
|
|
156
|
+
lengths[tid] = wp.length(points[tid])
|
|
157
|
+
|
|
158
|
+
|
|
159
|
+
# allocate an array of 3d points
|
|
160
|
+
points = wp.array(np.random.rand(num_points, 3), dtype=wp.vec3)
|
|
161
|
+
lengths = wp.zeros(num_points, dtype=float)
|
|
162
|
+
|
|
163
|
+
# launch kernel
|
|
164
|
+
wp.launch(kernel=length,
|
|
165
|
+
dim=len(points),
|
|
166
|
+
inputs=[points, lengths])
|
|
167
|
+
|
|
168
|
+
print(lengths)
|
|
169
|
+
```
|
|
170
|
+
|
|
171
|
+
## Running Notebooks
|
|
172
|
+
|
|
173
|
+
A few notebooks are available in the [notebooks](./notebooks/) directory to provide an overview over the key features available in Warp.
|
|
174
|
+
|
|
175
|
+
To run these notebooks, ``jupyterlab`` is required to be installed using:
|
|
176
|
+
|
|
177
|
+
```text
|
|
178
|
+
pip install jupyterlab
|
|
179
|
+
```
|
|
180
|
+
|
|
181
|
+
From there, opening the notebooks can be done with the following command:
|
|
182
|
+
|
|
183
|
+
```text
|
|
184
|
+
jupyter lab ./notebooks
|
|
185
|
+
```
|
|
186
|
+
|
|
187
|
+
* [Warp Core Tutorial: Basics](./notebooks/core_01_basics.ipynb) [](https://colab.research.google.com/github/NVIDIA/warp/blob/main/notebooks/core_01_basics.ipynb)
|
|
188
|
+
* [Warp Core Tutorial: Generics](./notebooks/core_02_generics.ipynb) [](https://colab.research.google.com/github/NVIDIA/warp/blob/main/notebooks/core_02_generics.ipynb)
|
|
189
|
+
* [Warp Core Tutorial: Points](./notebooks/core_03_points.ipynb) [](https://colab.research.google.com/github/NVIDIA/warp/blob/main/notebooks/core_03_points.ipynb)
|
|
190
|
+
* [Warp Core Tutorial: Meshes](./notebooks/core_04_meshes.ipynb) [](https://colab.research.google.com/github/NVIDIA/warp/blob/main/notebooks/core_04_meshes.ipynb)
|
|
191
|
+
* [Warp Core Tutorial: Volumes](./notebooks/core_05_volumes.ipynb) [](https://colab.research.google.com/github/NVIDIA/warp/blob/main/notebooks/core_05_volumes.ipynb)
|
|
192
|
+
* [Warp PyTorch Tutorial: Basics](./notebooks/pytorch_01_basics.ipynb) [](https://colab.research.google.com/github/NVIDIA/warp/blob/main/notebooks/pytorch_01_basics.ipynb)
|
|
193
|
+
* [Warp PyTorch Tutorial: Custom Operators](./notebooks/pytorch_02_custom_operators.ipynb) [](https://colab.research.google.com/github/NVIDIA/warp/blob/main/notebooks/pytorch_02_custom_operators.ipynb)
|
|
194
|
+
|
|
195
|
+
## Running Examples
|
|
196
|
+
|
|
197
|
+
The [warp/examples](./warp/examples/) directory contains a number of scripts categorized under subdirectories
|
|
198
|
+
that show how to implement various simulation methods using the Warp API.
|
|
199
|
+
Most examples will generate USD files containing time-sampled animations in the current working directory.
|
|
200
|
+
Before running examples, users should ensure that the ``usd-core``, ``matplotlib``, and ``pyglet`` packages are installed using:
|
|
201
|
+
|
|
202
|
+
```text
|
|
203
|
+
pip install warp-lang[extras]
|
|
204
|
+
```
|
|
205
|
+
|
|
206
|
+
These dependencies can also be manually installed using:
|
|
207
|
+
|
|
208
|
+
```text
|
|
209
|
+
pip install usd-core matplotlib pyglet
|
|
210
|
+
```
|
|
211
|
+
|
|
212
|
+
Examples can be run from the command-line as follows:
|
|
213
|
+
|
|
214
|
+
```text
|
|
215
|
+
python -m warp.examples.<example_subdir>.<example>
|
|
216
|
+
```
|
|
217
|
+
|
|
218
|
+
To browse the example source code, you can open the directory where the files are located like this:
|
|
219
|
+
|
|
220
|
+
```text
|
|
221
|
+
python -m warp.examples.browse
|
|
222
|
+
```
|
|
223
|
+
|
|
224
|
+
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.
|
|
225
|
+
|
|
226
|
+
USD files can be viewed or rendered inside [NVIDIA Omniverse](https://developer.nvidia.com/omniverse), Pixar's UsdView, and Blender. Note that Preview in macOS is not recommended as it has limited support for time-sampled animations.
|
|
227
|
+
|
|
228
|
+
Built-in unit tests can be run from the command-line as follows:
|
|
229
|
+
|
|
230
|
+
```text
|
|
231
|
+
python -m warp.tests
|
|
232
|
+
```
|
|
233
|
+
|
|
234
|
+
### warp/examples/core
|
|
235
|
+
|
|
236
|
+
<table>
|
|
237
|
+
<tbody>
|
|
238
|
+
<tr>
|
|
239
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_dem.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_dem.png"></a></td>
|
|
240
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_fluid.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_fluid.png"></a></td>
|
|
241
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_graph_capture.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_graph_capture.png"></a></td>
|
|
242
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_marching_cubes.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_marching_cubes.png"></a></td>
|
|
243
|
+
</tr>
|
|
244
|
+
<tr>
|
|
245
|
+
<td align="center">dem</td>
|
|
246
|
+
<td align="center">fluid</td>
|
|
247
|
+
<td align="center">graph capture</td>
|
|
248
|
+
<td align="center">marching cubes</td>
|
|
249
|
+
</tr>
|
|
250
|
+
<tr>
|
|
251
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_mesh.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_mesh.png"></a></td>
|
|
252
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_nvdb.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_nvdb.png"></a></td>
|
|
253
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_raycast.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_raycast.png"></a></td>
|
|
254
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_raymarch.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_raymarch.png"></a></td>
|
|
255
|
+
</tr>
|
|
256
|
+
<tr>
|
|
257
|
+
<td align="center">mesh</td>
|
|
258
|
+
<td align="center">nvdb</td>
|
|
259
|
+
<td align="center">raycast</td>
|
|
260
|
+
<td align="center">raymarch</td>
|
|
261
|
+
</tr>
|
|
262
|
+
<tr>
|
|
263
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_sample_mesh.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_sample_mesh.png"></a></td>
|
|
264
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_sph.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_sph.png"></a></td>
|
|
265
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_torch.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_torch.png"></a></td>
|
|
266
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/core/example_wave.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/core_wave.png"></a></td>
|
|
267
|
+
</tr>
|
|
268
|
+
<tr>
|
|
269
|
+
<td align="center">sample mesh</td>
|
|
270
|
+
<td align="center">sph</td>
|
|
271
|
+
<td align="center">torch</td>
|
|
272
|
+
<td align="center">wave</td>
|
|
273
|
+
</tr>
|
|
274
|
+
</tbody>
|
|
275
|
+
</table>
|
|
276
|
+
|
|
277
|
+
### warp/examples/fem
|
|
278
|
+
|
|
279
|
+
<table>
|
|
280
|
+
<tbody>
|
|
281
|
+
<tr>
|
|
282
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_diffusion_3d.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_diffusion_3d.png"></a></td>
|
|
283
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_mixed_elasticity.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_mixed_elasticity.png"></a></td>
|
|
284
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_apic_fluid.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_apic_fluid.png"></a></td>
|
|
285
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_streamlines.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_streamlines.png"></a></td>
|
|
286
|
+
</tr>
|
|
287
|
+
<tr>
|
|
288
|
+
<td align="center">diffusion 3d</td>
|
|
289
|
+
<td align="center">mixed elasticity</td>
|
|
290
|
+
<td align="center">apic fluid</td>
|
|
291
|
+
<td align="center">streamlines</td>
|
|
292
|
+
</tr>
|
|
293
|
+
<tr>
|
|
294
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_convection_diffusion.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_convection_diffusion.png"></a></td>
|
|
295
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_navier_stokes.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_navier_stokes.png"></a></td>
|
|
296
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_burgers.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_burgers.png"></a></td>
|
|
297
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/fem/example_magnetostatics.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/fem_magnetostatics.png"></a></td>
|
|
298
|
+
</tr>
|
|
299
|
+
<tr>
|
|
300
|
+
<td align="center">convection diffusion</td>
|
|
301
|
+
<td align="center">navier stokes</td>
|
|
302
|
+
<td align="center">burgers</td>
|
|
303
|
+
<td align="center">magnetostatics</td>
|
|
304
|
+
</tr>
|
|
305
|
+
</tbody>
|
|
306
|
+
</table>
|
|
307
|
+
|
|
308
|
+
### warp/examples/optim
|
|
309
|
+
|
|
310
|
+
<table>
|
|
311
|
+
<tbody>
|
|
312
|
+
<tr>
|
|
313
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_bounce.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_bounce.png"></a></td>
|
|
314
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_cloth_throw.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_cloth_throw.png"></a></td>
|
|
315
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_diffray.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_diffray.png"></a></td>
|
|
316
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_drone.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_drone.png"></a></td>
|
|
317
|
+
</tr>
|
|
318
|
+
<tr>
|
|
319
|
+
<td align="center">bounce</td>
|
|
320
|
+
<td align="center">cloth throw</td>
|
|
321
|
+
<td align="center">diffray</td>
|
|
322
|
+
<td align="center">drone</td>
|
|
323
|
+
</tr>
|
|
324
|
+
<tr>
|
|
325
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_inverse_kinematics.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_inverse_kinematics.png"></a></td>
|
|
326
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_spring_cage.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_spring_cage.png"></a></td>
|
|
327
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_trajectory.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_trajectory.png"></a></td>
|
|
328
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_softbody_properties.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_softbody_properties.png"></a></td>
|
|
329
|
+
</tr>
|
|
330
|
+
<tr>
|
|
331
|
+
<td align="center">inverse kinematics</td>
|
|
332
|
+
<td align="center">spring cage</td>
|
|
333
|
+
<td align="center">trajectory</td>
|
|
334
|
+
<td align="center">soft body properties</td>
|
|
335
|
+
</tr>
|
|
336
|
+
<tr>
|
|
337
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/optim/example_fluid_checkpoint.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/optim_fluid_checkpoint.png"></a></td>
|
|
338
|
+
<td></td>
|
|
339
|
+
<td></td>
|
|
340
|
+
<td></td>
|
|
341
|
+
</tr>
|
|
342
|
+
<tr>
|
|
343
|
+
<td align="center">fluid checkpoint</td>
|
|
344
|
+
<td align="center"></td>
|
|
345
|
+
<td align="center"></td>
|
|
346
|
+
<td align="center"></td>
|
|
347
|
+
</tr>
|
|
348
|
+
</tbody>
|
|
349
|
+
</table>
|
|
350
|
+
|
|
351
|
+
### warp/examples/sim
|
|
352
|
+
|
|
353
|
+
<table>
|
|
354
|
+
<tbody>
|
|
355
|
+
<tr>
|
|
356
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_cartpole.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_cartpole.png"></a></td>
|
|
357
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_cloth.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_cloth.png"></a></td>
|
|
358
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_granular.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_granular.png"></a></td>
|
|
359
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_granular_collision_sdf.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_granular_collision_sdf.png"></a></td>
|
|
360
|
+
</tr>
|
|
361
|
+
<tr>
|
|
362
|
+
<td align="center">cartpole</td>
|
|
363
|
+
<td align="center">cloth</td>
|
|
364
|
+
<td align="center">granular</td>
|
|
365
|
+
<td align="center">granular collision sdf</td>
|
|
366
|
+
</tr>
|
|
367
|
+
<tr>
|
|
368
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_jacobian_ik.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_jacobian_ik.png"></a></td>
|
|
369
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_quadruped.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_quadruped.png"></a></td>
|
|
370
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_rigid_chain.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_rigid_chain.png"></a></td>
|
|
371
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_rigid_contact.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_rigid_contact.png"></a></td>
|
|
372
|
+
</tr>
|
|
373
|
+
<tr>
|
|
374
|
+
<td align="center">jacobian ik</td>
|
|
375
|
+
<td align="center">quadruped</td>
|
|
376
|
+
<td align="center">rigid chain</td>
|
|
377
|
+
<td align="center">rigid contact</td>
|
|
378
|
+
</tr>
|
|
379
|
+
<tr>
|
|
380
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_rigid_force.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_rigid_force.png"></a></td>
|
|
381
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_rigid_gyroscopic.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_rigid_gyroscopic.png"></a></td>
|
|
382
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_rigid_soft_contact.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_rigid_soft_contact.png"></a></td>
|
|
383
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_soft_body.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_soft_body.png"></a></td>
|
|
384
|
+
</tr>
|
|
385
|
+
<tr>
|
|
386
|
+
<td align="center">rigid force</td>
|
|
387
|
+
<td align="center">rigid gyroscopic</td>
|
|
388
|
+
<td align="center">rigid soft contact</td>
|
|
389
|
+
<td align="center">soft body</td>
|
|
390
|
+
</tr>
|
|
391
|
+
<tr>
|
|
392
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/sim/example_cloth_self_contact.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/sim_example_cloth_self_contact.png"></a></td>
|
|
393
|
+
<td></td>
|
|
394
|
+
<td></td>
|
|
395
|
+
<td></td>
|
|
396
|
+
</tr>
|
|
397
|
+
<tr>
|
|
398
|
+
<td align="center">cloth self contact</td>
|
|
399
|
+
<td align="center"></td>
|
|
400
|
+
<td align="center"></td>
|
|
401
|
+
<td align="center"></td>
|
|
402
|
+
</tr>
|
|
403
|
+
</tbody>
|
|
404
|
+
</table>
|
|
405
|
+
|
|
406
|
+
### warp/examples/tile
|
|
407
|
+
|
|
408
|
+
<table>
|
|
409
|
+
<tbody>
|
|
410
|
+
<tr>
|
|
411
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/tile/example_tile_mlp.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/tile_mlp.png"></a></td>
|
|
412
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/tile/example_tile_nbody.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/tile_nbody.png"></a></td>
|
|
413
|
+
<td><a href="https://github.com/NVIDIA/warp/tree/main/warp/examples/tile/example_tile_walker.py"><img src="https://media.githubusercontent.com/media/NVIDIA/warp/refs/heads/main/docs/img/examples/tile_walker.png"></a></td>
|
|
414
|
+
</tr>
|
|
415
|
+
<tr>
|
|
416
|
+
<td align="center">mlp</td>
|
|
417
|
+
<td align="center">nbody</td>
|
|
418
|
+
<td align="center">walker</td>
|
|
419
|
+
</tr>
|
|
420
|
+
</tbody>
|
|
421
|
+
</table>
|
|
422
|
+
|
|
423
|
+
## Building
|
|
424
|
+
|
|
425
|
+
For developers who want to build the library themselves, the following tools are required:
|
|
426
|
+
|
|
427
|
+
* Microsoft Visual Studio 2019 upwards (Windows)
|
|
428
|
+
* GCC 9.4 upwards (Linux)
|
|
429
|
+
* CUDA Toolkit 11.5 or higher
|
|
430
|
+
* [Git LFS](https://git-lfs.github.com/) installed
|
|
431
|
+
|
|
432
|
+
After cloning the repository, users should run:
|
|
433
|
+
|
|
434
|
+
```text
|
|
435
|
+
python build_lib.py
|
|
436
|
+
```
|
|
437
|
+
|
|
438
|
+
Upon success, the script will output platform-specific binary files in `warp/bin/`.
|
|
439
|
+
The build script will look for the CUDA Toolkit in its default installation path.
|
|
440
|
+
This path can be overridden by setting the `CUDA_PATH` environment variable. Alternatively,
|
|
441
|
+
the path to the CUDA Toolkit can be passed to the build command as
|
|
442
|
+
`--cuda_path="..."`. After building, the Warp package should be installed using:
|
|
443
|
+
|
|
444
|
+
```text
|
|
445
|
+
pip install -e .
|
|
446
|
+
```
|
|
447
|
+
|
|
448
|
+
This ensures that subsequent modifications to the library will be reflected in the Python package.
|
|
449
|
+
|
|
450
|
+
## Learn More
|
|
451
|
+
|
|
452
|
+
Please see the following resources for additional background on Warp:
|
|
453
|
+
|
|
454
|
+
* [Product Page](https://developer.nvidia.com/warp-python)
|
|
455
|
+
* [SIGGRAPH 2024 Course Slides](https://dl.acm.org/doi/10.1145/3664475.3664543)
|
|
456
|
+
* [GTC 2024 Presentation](https://www.nvidia.com/en-us/on-demand/session/gtc24-s63345/)
|
|
457
|
+
* [GTC 2022 Presentation](https://www.nvidia.com/en-us/on-demand/session/gtcspring22-s41599)
|
|
458
|
+
* [GTC 2021 Presentation](https://www.nvidia.com/en-us/on-demand/session/gtcspring21-s31838)
|
|
459
|
+
* [SIGGRAPH Asia 2021 Differentiable Simulation Course](https://dl.acm.org/doi/abs/10.1145/3476117.3483433)
|
|
460
|
+
|
|
461
|
+
The underlying technology in Warp has been used in a number of research projects at NVIDIA including the following publications:
|
|
462
|
+
|
|
463
|
+
* Accelerated Policy Learning with Parallel Differentiable Simulation - Xu, J., Makoviychuk, V., Narang, Y., Ramos, F., Matusik, W., Garg, A., & Macklin, M. [(2022)](https://short-horizon-actor-critic.github.io)
|
|
464
|
+
* DiSECt: Differentiable Simulator for Robotic Cutting - Heiden, E., Macklin, M., Narang, Y., Fox, D., Garg, A., & Ramos, F [(2021)](https://github.com/NVlabs/DiSECt)
|
|
465
|
+
* gradSim: Differentiable Simulation for System Identification and Visuomotor Control - Murthy, J. Krishna, Miles Macklin, Florian Golemo, Vikram Voleti, Linda Petrini, Martin Weiss, Breandan Considine et al. [(2021)](https://gradsim.github.io)
|
|
466
|
+
|
|
467
|
+
## Frequently Asked Questions
|
|
468
|
+
|
|
469
|
+
See the [FAQ](https://nvidia.github.io/warp/faq.html) in the Warp documentation.
|
|
470
|
+
|
|
471
|
+
## Support
|
|
472
|
+
|
|
473
|
+
Problems, questions, and feature requests can be opened on [GitHub Issues](https://github.com/NVIDIA/warp/issues).
|
|
474
|
+
|
|
475
|
+
The Warp team also monitors the **#warp** forum on the public [Omniverse Discord](https://discord.com/invite/nvidiaomniverse) server, come chat with us!
|
|
476
|
+
|
|
477
|
+
For inquiries not suited for GitHub Issues or Discord, please email warp-python@nvidia.com.
|
|
478
|
+
|
|
479
|
+
## Versioning
|
|
480
|
+
|
|
481
|
+
Versions take the format X.Y.Z, similar to [Python itself](https://devguide.python.org/developer-workflow/development-cycle/#devcycle):
|
|
482
|
+
|
|
483
|
+
* Increments in X are reserved for major reworks of the project causing disruptive incompatibility (or reaching the 1.0 milestone).
|
|
484
|
+
* Increments in Y are for regular releases with a new set of features.
|
|
485
|
+
* Increments in Z are for bug fixes. In principle, there are no new features. Can be omitted if 0 or not relevant.
|
|
486
|
+
|
|
487
|
+
This is similar to [Semantic Versioning](https://semver.org/) but is less strict regarding backward compatibility.
|
|
488
|
+
Like with Python, some breaking changes can be present between minor versions if well-documented and gradually introduced.
|
|
489
|
+
|
|
490
|
+
Note that prior to 0.11.0, this schema was not strictly adhered to.
|
|
491
|
+
|
|
492
|
+
## License
|
|
493
|
+
|
|
494
|
+
Warp is provided under the Apache License, Version 2.0. Please see [LICENSE.md](./LICENSE.md) for full license text.
|
|
495
|
+
|
|
496
|
+
## Contributing
|
|
497
|
+
|
|
498
|
+
Contributions and pull requests from the community are welcome and are taken under the
|
|
499
|
+
terms described in the **Feedback** section of [LICENSE.md](LICENSE.md#9-feedback).
|
|
500
|
+
Please see the [Contribution Guide](https://nvidia.github.io/warp/modules/contribution_guide.html) for more
|
|
501
|
+
information on contributing to the development of Warp.
|
|
502
|
+
|
|
503
|
+
## Citing
|
|
504
|
+
|
|
505
|
+
If you use Warp in your research, please use the following citation:
|
|
506
|
+
|
|
507
|
+
```bibtex
|
|
508
|
+
@misc{warp2022,
|
|
509
|
+
title= {Warp: A High-performance Python Framework for GPU Simulation and Graphics},
|
|
510
|
+
author = {Miles Macklin},
|
|
511
|
+
month = {March},
|
|
512
|
+
year = {2022},
|
|
513
|
+
note= {NVIDIA GPU Technology Conference (GTC)},
|
|
514
|
+
howpublished = {\url{https://github.com/nvidia/warp}}
|
|
515
|
+
}
|
|
516
|
+
```
|