This is your work, valued
Assistant professor at Concordia University and Mila. Using information beyond the gradient to accelerate ML.
backpack. BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
615cockpit. Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
489unfoldNd. (N=1,2,3)-dimensional unfold (im2col) and fold (col2im) in PyTorch
109curvlinops. PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
69singd. [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)
24hbp. Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations
22phd-thesis. Source code for my PhD thesis: Backpropagation Beyond the Gradient
21einconv. Convolutions and more as einsum for PyTorch
18vivit. [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivatives & Newton steps
17sirfshampoo. [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
15phd-thesis-template. LaTeX template for my PhD thesis at the University of Tuebingen
15torch-jet. Taylor mode automatic differentiation (jets) in PyTorch
13kfac-tutorial. KFAC from scratch (KFS)---Paper & Code
10kfac-pinns-experiments. [NeurIPS2024] Paper and experiments for "Kronecker-Factored Approximate Curvature for Physics-Informed Neural Networks"
8org-export-setup. My org-export settings
3wandb_preempt. Code and tutorial on integrating wandb sweeps with Slurm pre-emption
2python-utilities. Python utility functions I often use
2backobs. Use DeepOBS with BackPACK
2vivit-experiments. Experiments for the TMLR 2023 paper "ViViT: Curvature Access Through the Generalized Gauss-Newton’s Low-rank Structure"
1