NERSC / nersc-dl-multigpu
single-GPU to multi-GPU training of PyTorch apps at NERSC
☆17Updated 10 months ago
Alternatives and similar repositories for nersc-dl-multigpu:
Users that are interested in nersc-dl-multigpu are comparing it to the libraries listed below
- SC23 Deep Learning at Scale Tutorial Material☆39Updated 5 months ago
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆10Updated last year
- PyTorch examples for NERSC systems☆30Updated 3 months ago
- Quasigeostrophic model in JAX (port of PyQG)☆19Updated 2 months ago
- explore the FV3 data for parameterization☆17Updated last month
- A tool to alleviate your Python package installation worries.☆19Updated last year
- Training material on writing machine learning code with PyTorch by ICCS☆28Updated 3 months ago
- SDSC Summer Institute 2022 material☆14Updated last year
- The ALCF hosts a regular simulation, data, and learning workshop to help users scale their applications. This repository contains the exa…☆60Updated 3 months ago
- Toolbox to use NNs in fortran☆12Updated 2 years ago
- Simple Xarray + JAX Integration☆16Updated 3 months ago
- Official implementation of Score-based Data Assimilation☆47Updated last year
- SDSC Summer Institute 2019 material☆18Updated 2 years ago
- Build scripts for PyTorch @ NERSC☆10Updated 8 months ago
- Examples for interactive distributed machine learning on the Cori supercomputer☆11Updated 5 years ago
- Lorenz 1996 two time-scale model for learning machine learning☆48Updated 2 weeks ago
- ☆20Updated 8 months ago
- ☆21Updated 4 years ago
- Simple differentiable approximate ocean models built with JAX.☆13Updated last year
- Pytorch bindings for Fortran☆87Updated last year
- Manipulating ragged arrays in an Array API compliant way.☆35Updated 3 weeks ago
- A library for directly calling PyTorch ML models from Fortran.☆96Updated this week
- Lua, Fortran and Python implementations of MAOOAM: the Modular Arbitrary-Order Ocean-Atmosphere Model☆22Updated 3 years ago
- Unleash the true power of scheduling☆28Updated 2 weeks ago
- A package to train machine learning models on geospatial data, mainly for weather and climate. Used to run ArchesWeather and ArchesWeathe…☆27Updated this week
- AI Training Series Material☆29Updated 4 months ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆28Updated last week
- ☆35Updated this week
- PyTorch training at CSCS☆15Updated 11 months ago
- Lecture and hands-on material for Track 8- Machine Learning of Argonne Training Program on Extreme-Scale Computing☆36Updated 6 months ago