NERSC / nersc-dl-wandb
Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines
☆10Updated last year
Alternatives and similar repositories for nersc-dl-wandb:
Users that are interested in nersc-dl-wandb are comparing it to the libraries listed below
- PyTorch examples for NERSC systems☆30Updated 2 months ago
- Repo for a DOE HPC workflow training event☆13Updated last year
- SC23 Deep Learning at Scale Tutorial Material☆38Updated 4 months ago
- CUDA implementations of MACE models☆11Updated last week
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆23Updated last month
- Equivariant machine learning interatomic potentials in JAX.☆66Updated last year
- ☆37Updated 5 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆96Updated 2 months ago
- Steerable E(3) GNN in jax☆24Updated last year
- Tools for building equivariant polynomials on reductive Lie groups.☆28Updated last year
- Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.☆82Updated this week
- ☆13Updated last year
- Implementation of the Euclidean fast attention (EFA) algorithm☆32Updated last month
- MESS: Modern Electronic Structure Simulations☆24Updated 3 months ago
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆12Updated last year
- MESS: Modern Electronic Structure Simulations☆18Updated 3 months ago
- ☆21Updated 9 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆33Updated last year
- Library for steering campaigns of simulations on supercomputers☆49Updated 2 weeks ago
- Implementation of various equivariant models in JAX☆12Updated 9 months ago
- Point Edge Transformer☆19Updated 2 months ago
- E(n) Equivariant GNN in jax☆11Updated last year
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆57Updated this week
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆162Updated this week
- Pure C implementation of e3nn☆16Updated last month
- Exchange correlation functionals translated from libxc to jax☆43Updated last year
- ☆48Updated 2 years ago
- tools for graph-based machine-learning potentials in jax☆18Updated 9 months ago
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆54Updated 3 weeks ago