eth-cscs / pytorch-training
PyTorch training at CSCS
☆15Updated 10 months ago
Alternatives and similar repositories for pytorch-training:
Users that are interested in pytorch-training are comparing it to the libraries listed below
- Multi-GPU training with TensorFlow on Piz Daint☆12Updated 3 years ago
- Pytorch bindings for Fortran☆86Updated last year
- PyTorch examples for NERSC systems☆30Updated 3 months ago
- ☆33Updated 3 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆23Updated 2 months ago
- Making Fabio Crameri's perceptually uniform colourmaps for geosciences available on PyPI and conda-forge☆142Updated this week
- JAX bindings for the NVIDIA cuDecomp library☆32Updated this week
- Official implementation of Score-based Data Assimilation☆45Updated last year
- ☆31Updated 4 years ago
- Things that make me feel productive☆15Updated 2 years ago
- explore the FV3 data for parameterization☆17Updated 3 weeks ago
- Stencil computations in JAX☆70Updated last year
- Custom LaTeX2e documentclass for typesetting beautiful, modern theses.☆9Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆114Updated this week
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆10Updated last year
- Bootcamp notebooks☆51Updated 8 months ago
- SC23 Deep Learning at Scale Tutorial Material☆38Updated 4 months ago
- A Python toolkit for coordinating asynchronous and dynamic ensembles of calculations.☆66Updated this week
- Python library for generating high-performance implementations of stencil kernels for weather and climate modeling from a domain-specific…☆113Updated this week
- Simple Xarray + JAX Integration☆16Updated 2 months ago
- Finite difference tools in JAX☆17Updated 11 months ago
- JAX + Units☆44Updated this week
- Orthogonal polynomials with JAX☆12Updated this week
- SciML Benchmarking Suite for AI for Science☆38Updated 6 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆84Updated 3 weeks ago
- The default repo for wiki and issues relating to the AI4ER CDT☆9Updated 4 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 4 months ago
- High Performance Computing for Weather and Climate☆37Updated 4 months ago
- A library for directly calling PyTorch ML models from Fortran.☆91Updated this week
- Fully and Partially Bayesian Neural Nets☆46Updated this week