saforem2 / ezpzLinks
Train across all your devices, ezpz 🍋
☆23Updated 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☆24Updated last week
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆62Updated this week
- ☆59Updated 2 years ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆85Updated last week
- A parallel framework for training deep neural networks☆63Updated 4 months ago
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆13Updated 2 years ago
- AI Training Series Material☆37Updated 10 months ago
- Running Jax in PyTorch Lightning☆109Updated 7 months ago
- Graphium: Scaling molecular GNNs to infinity.☆233Updated 3 months ago
- Collection of small examples for running on ALCF resources☆19Updated 2 weeks ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆269Updated last week
- SciML Benchmarking Suite for AI for Science☆42Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆16Updated 2 weeks ago
- Graph neural networks in JAX.☆67Updated last year
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆98Updated last month
- Code for paper E(n)-Equivariant Topological Neural Networks☆37Updated 8 months ago
- ☆224Updated 8 months ago
- The OGB-LSC is the Large Scale Competition by Open Graph Benchmark to help accelerate research into machine learning on graph structured …☆79Updated last year
- Material for the SC21 Deep Learning at Scale Tutorial☆26Updated 2 years ago
- An Online Deep Learning Interface for HPC programs on NVIDIA GPUs☆169Updated 3 weeks ago
- ☆125Updated 8 months ago
- SC23 Deep Learning at Scale Tutorial Material☆46Updated 10 months ago
- [ICLR 2022] Data-Efficient Graph Grammar Learning for Molecular Generation☆96Updated last year
- A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/☆31Updated 4 months ago
- ☆43Updated 3 weeks ago
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆35Updated 4 years ago
- Modalities, a PyTorch-native framework for distributed and reproducible foundation model training.☆85Updated this week
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆131Updated 3 months ago
- Intel Gaudi's Megatron DeepSpeed Large Language Models for training☆13Updated 7 months ago