NERSC / nersc-dl-multigpu
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☆12Updated last year
- SC23 Deep Learning at Scale Tutorial Material☆44Updated 7 months ago
- PyTorch examples for NERSC systems☆32Updated 6 months ago
- Deep Learning at Scale Training Event at NERSC☆18Updated 2 months ago
- ☆37Updated 9 months ago
- Simple Xarray + JAX Integration☆18Updated 6 months ago
- differentiable (binned) likelihoods with JAX☆22Updated this week
- Manipulating ragged arrays in an Array API compliant way.☆41Updated 2 weeks ago
- Quasigeostrophic model in JAX (port of PyQG)☆19Updated 5 months ago
- A library for directly calling PyTorch ML models from Fortran.☆115Updated this week
- Helper script for executing commands before connecting to remote.☆37Updated 8 months ago
- Materials for a workshop focusing on https://jobqueue.dask.org☆28Updated 5 years ago
- Pytorch bindings for Fortran☆93Updated last year
- ☆27Updated last week
- PyTorch training at CSCS☆15Updated last year
- ☆13Updated last year
- Unleash the true power of scheduling☆29Updated last month
- Simple differentiable approximate ocean models built with JAX.☆14Updated last year
- Example applications for the Department of Energy Computational Science Graduate Fellowship☆16Updated last week
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆86Updated last week
- Efficient Differentiable n-d PDE solvers in JAX.☆37Updated 6 months ago
- SDSC Summer Institute 2019 material☆18Updated 2 years ago
- Official implementation of Score-based Data Assimilation☆60Updated last year
- Linux productivity tools and practices for researchers☆82Updated this week
- AI Training Series Material☆35Updated 7 months ago
- SDSC Summer Institute 2022 material☆14Updated last year
- Up-scale python functions for high performance computing (HPC)☆31Updated this week
- Fully and Partially Bayesian Neural Nets☆68Updated last week
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆20Updated 2 weeks ago
- Simulation-Enabled Prediction, Inference, and Analysis: physics-informed statistical learning.☆35Updated last year