saforem2 / ezpzLinks
Train across all your devices, ezpz 🍋
☆21Updated this week
Alternatives and similar repositories for ezpz
Users that are interested in ezpz are comparing it to the libraries listed below
Sorting:
- This is a repository with examples to run inference endpoints on various ALCF clusters☆20Updated last week
- SC23 Deep Learning at Scale Tutorial Material☆45Updated 8 months ago
- ☆55Updated last year
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated 2 months ago
- ☆37Updated 10 months ago
- Code for paper E(n)-Equivariant Topological Neural Networks☆35Updated 6 months ago
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆81Updated last week
- Library for steering campaigns of simulations on supercomputers☆55Updated this week
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆53Updated 3 weeks ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆17Updated 4 months ago
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆13Updated last year
- Cosmic Tagging Network for Neutrino Physics☆13Updated 11 months ago
- Call for participation in the impact of LLM for scientific discovery☆71Updated last year
- Collection of small examples for running on ALCF resources☆19Updated last week
- FLOPS counter for all your GPU benchmarking needs☆13Updated 9 months ago
- The OGB-LSC is the Large Scale Competition by Open Graph Benchmark to help accelerate research into machine learning on graph structured …☆77Updated 10 months ago
- jax library for E3 Equivariant Neural Networks☆207Updated 4 months ago
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆217Updated last week
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆221Updated last week
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated last year
- Minimal AlphaFold☆53Updated 9 months ago
- Tutorial notebooks for SciFM24☆11Updated last year
- Official repository for discrete Walk-Jump Sampling (dWJS)☆53Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆35Updated 3 months ago
- Recipes for software stacks on Alps vClusters.☆13Updated this week
- [TMLR 2023] Simulate time-integrated coarse-grained MD with multi-scale graph neural networks☆71Updated last year
- An implementation of ESM2 in Equinox+JAX☆27Updated this week
- A JAX library for Density Functional Theory.☆53Updated this week
- ☆36Updated last month