warp
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
Train, fine-tune, and evaluate models — the infrastructure layer.
Key features
- Python 写 GPU kernel
- JIT 编译高效执行
- kernel 可微分
- 物理仿真 primitives
- 接入 PyTorch/JAX
- CPU/GPU 通跑
Good for
- 物理仿真
- 机器人仿真
- 可微分编程
- GPU 加速计算
Stack & tags
PythonCUDAPyTorchJAXPyPIcudadifferentiable-programminggpugpu-accelerationnvidianvidia-warppython
Quick start
pip install warp-langFrom the project README on GitHub.
Similar open-source projects
tensorflow
★ 200,721An Open Source Machine Learning Framework for Everyone
transformers
★ 167,239🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.