e3nn_mlx.radius_graph¶

e3nn_mlx.radius_graph(positions, radius, batch=None)[source]¶

Build the directed, loop-free batched radius graph used by gate-points models.

Topology construction is intentionally eager: MLX 0.31 has no device-side dynamic nonzero operation. Gradients should flow through edge vectors, not through the discrete neighbor selection itself.