ShashankSubramanian

nersc-dl-wandb

Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines

sc24-dl-tutorial

SC24 Deep Learning at Scale Tutorial Material

adaptive-selfsupervision-pinns

Gradient-based adaptive sampling algorithms for self-supervising PINNs

DDP_UNet

3D U-Net for pytorch DDP training

DPT

Dense Prediction Transformers

EllipticSolvers

FourCastNet

Initial public release of code, data, and model weights for FourCastNet

GLIA

GLioblastoma Image Analysis for integrating brain tumor growth models with medical imaging

neuraloperators-TL-scaling

Code for Characterizing Scaling and Transfer Learning Behavior of FNO in SciML

transformer-perf-estimates

Performance Estimates for Transformer AI Models in Science

ShashankSubramanian/neuraloperators-TL-scaling

Code for Characterizing Scaling and Transfer Learning Behavior of FNO in SciML

ShashankSubramanian/adaptive-selfsupervision-pinns

Gradient-based adaptive sampling algorithms for self-supervising PINNs

ShashankSubramanian/GLIA

GLioblastoma Image Analysis for integrating brain tumor growth models with medical imaging

ShashankSubramanian/transformer-perf-estimates

Performance Estimates for Transformer AI Models in Science

ShashankSubramanian/DDP_UNet

3D U-Net for pytorch DDP training

ShashankSubramanian/DPT

Dense Prediction Transformers

ShashankSubramanian/EllipticSolvers

ShashankSubramanian/FourCastNet

Initial public release of code, data, and model weights for FourCastNet

ShashankSubramanian/modulus-launch-fork

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems

ShashankSubramanian/pytorch-examples

PyTorch examples for NERSC systems

ShashankSubramanian/ssl-sky-surveys

ShashankSubramanian/ViT-pytorch

Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)