Max Zhdanov

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

clifford-group-equivariant-cnns. Clifford-Steerable Convolutional Neural Networks [ICML'24]

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

flash-clifford. ⚡ Triton implementation of Clifford algebra neural networks.

implicit-steerable-kernels. Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]

EEGCN. Investigating Brain Connectivity with Graph Neural Networks and GNNExplainer [ICPR'22]

jax-geometric. Implementation of various equivariant models in JAX

eegVAE. Variational Inference on EEG data

adsgnn. AdS-GNN - a Conformally Equivariant Graph Neural Network

gisaxs-reconstruction. Simulation-based inference for GISAXS data

balltree. A fast, parallel ball tree implementation with PyTorch integration, built using Cython and OpenMP.

GNNExplainer. GNNExplainer extension to calculate explanation for graph classification