Operator Registration
=====================

.. contents::
   :local:
   :depth: 1



General Operator Customization Interface
----------------------------------------

.. currentmodule:: brainpy.math
.. automodule:: brainpy.math

.. autosummary::
   :toctree: generated/
   :nosignatures:
   :template: classtemplate.rst

   XLACustomOp



CPU Operator Customization with Taichi
-------------------------------------

.. currentmodule:: brainpy.math
.. automodule:: brainpy.math

.. autosummary::
   :toctree: generated/

   clear_taichi_aot_caches
   count_taichi_aot_kernels






CPU Operator Customization with Numba
-------------------------------------

.. currentmodule:: brainpy.math
.. automodule:: brainpy.math

.. autosummary::
   :toctree: generated/
   :nosignatures:
   :template: classtemplate.rst

   CustomOpByNumba


.. autosummary::
   :toctree: generated/

   register_op_with_numba
   compile_cpu_signature_with_numba



Operator Autograd Customization
-------------------------------

.. currentmodule:: brainpy.math
.. automodule:: brainpy.math

.. autosummary::
   :toctree: generated/

   defjvp