需要点券:30
下载地址:DiffTaichi:物理模拟的微分编程
文件格式:PDF
文件大小:3.8M,20页
文件节选:我们介绍了 DiffTaichi,一种新的可微分编程语言,专为构建高性能可微分物理模拟器而设计。 基于命令式编程语言,DiffTaichi 使用保留算术强度和并行性的源代码转换生成模拟步骤的梯度。 轻量级磁带用于记录整个模拟程序结构并以相反的顺序重放梯度内核,以进行端到端的反向传播。 我们在 10 个不同的物理模拟器上展示了我们的语言在基于梯度的学习和优化任务中的性能和生产力。 例如,用我们的语言编写的可微弹性对象模拟器比手工设计的 CUDA 版本短 4.2 倍,但运行速度却一样快,比 TensorFlow 实现快 188 倍。 使用我们的可微程序,神经网络控制器通常仅在数十次迭代内即可优化。