warp.jax\_kernel
================

.. currentmodule:: warp

.. autofunction:: jax_kernel