NERSC / nersc-dl-multigpu
single-GPU to multi-GPU training of PyTorch apps at NERSC
☆18Updated 11 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
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆11Updated last year
- SC23 Deep Learning at Scale Tutorial Material☆41Updated 6 months ago
- PyTorch examples for NERSC systems☆31Updated 5 months ago
- Unleash the true power of scheduling☆30Updated 2 weeks ago
- JAX bindings for the NVIDIA cuDecomp library☆32Updated last month
- differentiable (binned) likelihoods with JAX☆21Updated last week
- A simple JAX solver for the 2D Euler equations for benchmarking☆20Updated 5 months ago
- Example applications for the Department of Energy Computational Science Graduate Fellowship☆16Updated 8 months ago
- ☆21Updated 4 years ago
- Training material on writing machine learning code with PyTorch by ICCS☆28Updated last week
- My CV☆37Updated this week
- Deep Learning at Scale Training Event at NERSC☆18Updated 3 weeks ago
- A library for directly calling PyTorch ML models from Fortran.☆107Updated this week
- SDSC Summer Institute 2022 material☆14Updated last year
- Quasigeostrophic model in JAX (port of PyQG)☆19Updated 3 months ago
- The ALCF hosts a regular simulation, data, and learning workshop to help users scale their applications. This repository contains the exa…☆61Updated 4 months ago
- Official TensorFlow 2.0 tutorial notebooks for the Deep Learning for Science School at LBNL☆44Updated 5 years ago
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆20Updated 9 months ago
- Pytorch bindings for Fortran☆91Updated last year
- Learning materials for scientific software development☆83Updated 3 weeks ago
- JAX + Units☆47Updated 2 months ago
- ☆13Updated last year
- Manipulating ragged arrays in an Array API compliant way.☆41Updated 3 weeks ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆27Updated 4 months ago
- Simple Xarray + JAX Integration☆16Updated 5 months ago
- ☆37Updated 8 months ago
- Simulation-based inference in JAX☆31Updated last month
- ☆21Updated last year
- Examples for interactive distributed machine learning on the Cori supercomputer☆11Updated 5 years ago