colossalai.kernel.jit
=====================

.. automodule:: colossalai.kernel.jit
   :members:


.. toctree::
   :maxdepth: 2

   colossalai.kernel.jit.bias_dropout_add
   colossalai.kernel.jit.bias_gelu
   colossalai.kernel.jit.option