xlstm_jax.kernels.kernel_utils
==============================

.. py:module:: xlstm_jax.kernels.kernel_utils


Attributes
----------

.. autoapisummary::

   xlstm_jax.kernels.kernel_utils._jax_to_triton_dtype


Functions
---------

.. autoapisummary::

   xlstm_jax.kernels.kernel_utils.jax2triton_dtype


Module Contents
---------------

.. py:data:: _jax_to_triton_dtype

.. py:function:: jax2triton_dtype(dtype)

   Converts a JAX dtype to a Triton dtype.

   :param dtype: JAX dtype.

   :returns: Triton dtype.