NERSC / pytorch-examples
PyTorch examples for NERSC systems
☆29Updated last week
Related projects ⓘ
Alternatives and complementary repositories for pytorch-examples
- Material for the SC22 Deep Learning at Scale Tutorial☆39Updated last year
- SC23 Deep Learning at Scale Tutorial Material☆37Updated last month
- Things that make me feel productive☆15Updated 2 years ago
- For developing and reproducing ML + HEP projects.☆22Updated this week
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 2 weeks ago
- differentiable (binned) likelihoods with JAX☆18Updated last week
- ☆13Updated last year
- ☆21Updated last year
- Tutorial Code for MLHEP pyprob☆18Updated last year
- ☆31Updated 4 years ago
- PyTorch training at CSCS☆15Updated 7 months ago
- Unleash the true power of scheduling☆24Updated 3 weeks ago
- single-GPU to multi-GPU training of PyTorch apps at NERSC☆15Updated 6 months ago
- Fully and Partially Bayesian Neural Nets☆22Updated this week
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated last year
- Differentiable and accelerated spherical transforms with JAX☆133Updated 2 weeks ago
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- Jet tagging in the Lund plane with graph networks☆8Updated 2 years ago
- Third version of larcv. This is a complete replacement for larcv2.☆11Updated 4 months ago
- Official TensorFlow 2.0 tutorial notebooks for the Deep Learning for Science School at LBNL☆42Updated 5 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated this week
- The ALCF hosts a regular simulation, data, and learning workshop to help users scale their applications. This repository contains the exa…☆58Updated last week
- ☆19Updated 2 years ago
- ALCF Computational Performance Workshop☆34Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆11Updated 8 months ago
- Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics☆45Updated 10 months ago
- Code to estimate Renormalized Mutual Information in simple settings☆12Updated 3 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 4 months ago
- ☆19Updated last year