NERSC / nersc-dl-multigpu
single-GPU to multi-GPU training of PyTorch apps at NERSC
☆15Updated 7 months ago
Related projects ⓘ
Alternatives and complementary repositories for nersc-dl-multigpu
- SC23 Deep Learning at Scale Tutorial Material☆37Updated 2 months ago
- PyTorch examples for NERSC systems☆29Updated 3 weeks ago
- Pytorch bindings for Fortran☆85Updated 10 months ago
- SDSC Summer Institute 2019 material☆18Updated last year
- A simple JAX solver for the 2D Euler equations for benchmarking☆18Updated last month
- Lecture and hands-on material for Track 8- Machine Learning of Argonne Training Program on Extreme-Scale Computing☆33Updated 3 months ago
- Experimental pipeline prototype software☆13Updated last month
- Official implementation of Score-based Data Assimilation☆45Updated 10 months ago
- Example applications for the Department of Energy Computational Science Graduate Fellowship☆12Updated 4 months ago
- The ALCF hosts a regular simulation, data, and learning workshop to help users scale their applications. This repository contains the exa…☆59Updated 2 weeks ago
- Examples for interactive distributed machine learning on the Cori supercomputer☆11Updated 5 years ago
- A library for directly calling PyTorch ML models from Fortran.☆78Updated this week
- ☆21Updated 4 years ago
- Quasigeostrophic model in JAX (port of PyQG)☆18Updated last month
- Official TensorFlow 2.0 tutorial notebooks for the Deep Learning for Science School at LBNL☆42Updated 5 years ago
- A library for directly calling TensorFlow / Keras ML models from Fortran.☆30Updated last year
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆27Updated this week
- A tool to alleviate your Python package installation worries.☆19Updated last year
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆21Updated 10 months ago
- ☆65Updated 2 years ago
- ALCF Computational Performance Workshop☆34Updated 2 years ago
- Learning materials for scientific software development☆78Updated this week
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 3 weeks ago
- Simple Xarray + JAX Integration☆13Updated 3 weeks ago
- Large-Scale Multimodal Dataset of Astronomical Data☆41Updated this week
- differentiable (binned) likelihoods with JAX☆18Updated last week
- Dynamic dispatch of a list of command-line tasks, locally or on a cluster. Supports retrying failed tasks, and adding/removing compute re…☆39Updated last week
- Unleash the true power of scheduling☆24Updated this week
- A single column model of atmospheric boundary layer turbulence and clouds☆11Updated 4 years ago
- Statistics of Turbulence Python Package☆61Updated 2 months ago