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

.. automodule:: colossalai.kernel
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.kernel.cuda_native
   colossalai.kernel.jit