saforem2 / ezpz
Train across all your devices, ezpz 🍋
☆20Updated last 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☆19Updated last week
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated 2 months ago
- ☆36Updated 3 weeks ago
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆12Updated last year
- AI Training Series Material☆35Updated 7 months ago
- Collection of small examples for running on ALCF resources☆18Updated last week
- Cosmic Tagging Network for Neutrino Physics☆13Updated 10 months ago
- ☆55Updated last year
- SC23 Deep Learning at Scale Tutorial Material☆44Updated 8 months ago
- Library for steering campaigns of simulations on supercomputers☆53Updated 2 weeks ago
- Tutorial notebooks for SciFM24☆11Updated last year
- SciML Benchmarking Suite for AI for Science☆40Updated 9 months ago
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆80Updated this week
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- The OGB-LSC is the Large Scale Competition by Open Graph Benchmark to help accelerate research into machine learning on graph structured …☆76Updated 9 months ago
- ALCF Computational Performance Workshop☆37Updated 2 years ago
- ☆37Updated 9 months ago
- scalable data movement in Exascale Supercomputers☆15Updated 3 weeks ago
- Python package template☆15Updated this week
- GenSLMs: Genome-scale language models reveal SARS-CoV-2 evolutionary dynamics☆128Updated 8 months ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated last year
- The ALCF hosts a regular simulation, data, and learning workshop to help users scale their applications. This repository contains the exa…☆63Updated 6 months ago
- Continuous Spatiotemporal Transformer☆20Updated last year
- ☆59Updated last year
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆51Updated last week
- Minimal AlphaFold☆53Updated 8 months ago
- Running Jax in PyTorch Lightning☆100Updated 5 months ago
- [TMLR 2023] Simulate time-integrated coarse-grained MD with multi-scale graph neural networks☆70Updated last year
- This Denoising Force Field (DFF) codebase provides a Pytorch framework for the method presented in Two for one: Diffusion models and forc…☆60Updated 11 months ago
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆191Updated this week