def METHOD_NAME(self, edges):
    # Transform from Bessel basis to dense vector
    rbf = self.dense_rbf1(edges.data["rbf"])
    rbf = self.dense_rbf2(rbf)
    # Initial transformation
    x_ji = self.dense_ji(edges.data["m"])
    x_kj = self.dense_kj(edges.data["m"])
    if self.activation is not None:
        x_ji = self.activation(x_ji)
        x_kj = self.activation(x_kj)
    x_kj = self.down_projection(x_kj * rbf)
    if self.activation is not None:
        x_kj = self.activation(x_kj)
    return {"x_kj": x_kj, "x_ji": x_ji}