NERSC / nersc-dl-multigpuLinks
single-GPU to multi-GPU training of PyTorch apps at NERSC
☆18Updated last year
Alternatives and similar repositories for nersc-dl-multigpu
Users that are interested in nersc-dl-multigpu are comparing it to the libraries listed below
Sorting:
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆13Updated last year
- PyTorch examples for NERSC systems☆32Updated 7 months ago
- SC23 Deep Learning at Scale Tutorial Material☆45Updated 9 months ago
- Unleash the true power of scheduling☆29Updated 2 months ago
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Updated 8 months ago
- Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.☆56Updated last year
- Deep Learning at Scale Training Event at NERSC☆19Updated 3 weeks ago
- Simple Xarray + JAX Integration☆19Updated 7 months ago
- Training material on writing machine learning code with PyTorch by ICCS☆31Updated this week
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆20Updated last month
- JAX bindings for the NVIDIA cuDecomp library☆37Updated 2 weeks ago
- Statistics of Turbulence Python Package☆66Updated 5 months ago
- ☆37Updated 11 months ago
- Examples for interactive distributed machine learning on the Cori supercomputer☆11Updated 5 years ago
- SDSC Summer Institute 2019 material☆18Updated 2 years ago
- Quasigeostrophic model in JAX (port of PyQG)☆19Updated 3 weeks ago
- Simulation-based inference in JAX☆32Updated 2 months ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆31Updated this week
- Example applications for the Department of Energy Computational Science Graduate Fellowship☆16Updated last month
- A library for directly calling TensorFlow / Keras ML models from Fortran.☆32Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Updated last year
- Official TensorFlow 2.0 tutorial notebooks for the Deep Learning for Science School at LBNL☆43Updated 5 years ago
- Official implementation of Score-based Data Assimilation☆64Updated last year
- Simulation-Enabled Prediction, Inference, and Analysis: physics-informed statistical learning.☆35Updated last year
- Easily generate random fields☆19Updated last month
- My CV☆37Updated this week
- ☆13Updated last year
- PyTorch training at CSCS☆15Updated 2 weeks ago
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated 8 months ago
- AI Training Series Material☆37Updated 8 months ago