=================
Dimenet operators
=================

.. currentmodule:: pylibcugraphops

Radial Basis Functions
----------------------
.. autosummary::
   :toctree: ../../api/ops

   dimenet.radial_basis_fwd
   dimenet.radial_basis_bwd
   dimenet.radial_basis_bwd_bwd

Edge-to-Edge Aggregation
-------------------------
.. autosummary::
   :toctree: ../../api/ops

   dimenet.agg_edge_to_edge_fwd
   dimenet.agg_edge_to_edge_bwd
   dimenet.agg_edge_to_edge_bwd2_grad
   dimenet.agg_edge_to_edge_bwd2_main