================= 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