(Sparse) Attention to the Details: Preserving Spectral Fidelity in ML-based Weather Forecasting Models [ICML'26]

⚡ Triton implementation of Clifford algebra neural networks.

Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]

Clifford-Steerable Convolutional Neural Networks [ICML'24]

Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]

Neural network based solvers for partial differential equations and inverse problems . Implementation of physics-informed neural networks in pytorch.

Implementation of various equivariant models in JAX