saforem2 / ezpzLinks
Train across all your devices, ezpz 🍋
☆24Updated 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☆26Updated last week
- ☆61Updated 2 years ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆17Updated last week
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆13Updated 2 years ago
- SciML Benchmarking Suite for AI for Science☆42Updated last year
- Running Jax in PyTorch Lightning☆114Updated 11 months ago
- ☆234Updated last week
- A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/☆35Updated 3 weeks ago
- Collection of small examples for running on ALCF resources☆19Updated 3 months ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆65Updated last month
- ☆47Updated 4 months ago
- Proxy server to Argo API, OpenAI format compatible☆15Updated last week
- Parallel framework for training and fine-tuning deep neural networks☆68Updated last week
- Cosmic Tagging Network for Neutrino Physics☆13Updated last year
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆94Updated 2 weeks ago
- SC23 Deep Learning at Scale Tutorial Material☆48Updated last year
- ☆33Updated last year
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆38Updated 4 years ago
- AI Training Series Material☆38Updated last month
- SC24 Deep Learning at Scale Tutorial Material☆33Updated 9 months ago
- ALCF Computational Performance Workshop☆38Updated 3 years ago
- Material for the SC21 Deep Learning at Scale Tutorial☆27Updated 2 years ago
- ☆61Updated last year
- Einsum-like high-level array sharding API for JAX☆34Updated last year
- CSCS public documentation☆23Updated last week
- Sparsity support for PyTorch☆37Updated 7 months ago
- PyTorch examples for NERSC systems☆34Updated last year
- Library for steering campaigns of simulations on supercomputers☆58Updated 5 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆149Updated last week