Haoliang Wang

Speeding up, GPU and 3D

3

HI Filipe, any progress on speeding up the engine? And by the way, do you have GPU and 3D version of the physics engine? Thanks!

[BUG] Error when using jax_callable with vmap

2

### Bug Description I'm trying to use Warp's JAX FFI feature, I see that the `jax_callback` feature supports [vmap](https://nvidia.github.io/warp/modules/interoperability.html#warp.jax_experimental.ffi.jax_callable). However, when I try to use it, it complains about dimensions....