saforem2 / ezpz
Train across all your devices, ezpz 🍋
☆20Updated this week
Alternatives and similar repositories for ezpz:
Users that are interested in ezpz are comparing it to the libraries listed below
- This is a repository with examples to run inference endpoints on various ALCF clusters☆19Updated last week
- Adaptive Parallel PDF Parsing and Resource Scaling Engine☆12Updated last month
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆17Updated 3 months ago
- SC23 Deep Learning at Scale Tutorial Material☆43Updated 7 months ago
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆11Updated last year
- ☆55Updated last year
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated last month
- Minimal AlphaFold☆53Updated 8 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems☆45Updated last week
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆32Updated last month
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆79Updated this week
- Code for paper E(n)-Equivariant Topological Neural Networks☆34Updated 4 months ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Python library for analyzing the internal structure of deep neural networks.☆18Updated 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
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆49Updated last month
- ☆36Updated 2 months ago
- Code corresponding to the paper Diffusion Earth Mover's Distance and Distribution Embeddings☆36Updated 4 months ago
- Official implementation of Fisher-Flow Matching (NeurIPS 2024).☆21Updated 6 months ago
- Library for steering campaigns of simulations on supercomputers☆53Updated 2 weeks 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
- Cosmic Tagging Network for Neutrino Physics☆13Updated 10 months ago
- [TMLR 2023] Simulate time-integrated coarse-grained MD with multi-scale graph neural networks☆70Updated last year
- ☆221Updated 5 months ago
- Tutorial notebooks for SciFM24☆11Updated last year
- ICML2024: Equivariant Graph Neural Operator for Modeling 3D Dynamics☆56Updated last year
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- Official repository for the Topological Deep Learning Challenge 2024, organized by TAG-DS & PyT-Team and hosted by GRaM Workshop @ ICML 2…☆38Updated 2 months ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆39Updated last year
- A collection of graph neural networks implementations in JAX☆32Updated last year