saforem2 / ezpzLinks
Train across all your devices, ezpz 🍋
☆24Updated 2 weeks ago
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 3 weeks ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆66Updated last month
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆17Updated 3 weeks ago
- Proxy server to Argo API, OpenAI format compatible☆15Updated last month
- ☆61Updated 2 years ago
- SciML Benchmarking Suite for AI for Science☆42Updated last year
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- Running Jax in PyTorch Lightning☆115Updated 11 months ago
- ☆47Updated 4 months ago
- Experimental paper writing linter.☆35Updated last year
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆23Updated 8 months ago
- ☆236Updated 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, …☆39Updated 4 years ago
- Cosmic Tagging Network for Neutrino Physics☆13Updated last year
- A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/☆35Updated last month
- Collection of small examples for running on ALCF resources☆19Updated 4 months ago
- AI Training Series Material☆38Updated 2 months ago
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆95Updated this week
- FLOPS counter for all your GPU benchmarking needs☆13Updated last year
- Parallel framework for training and fine-tuning deep neural networks☆70Updated last month
- Efficient LLM inference on Slurm clusters using vLLM.☆86Updated this week
- Jupyter notebooks, jobscripts and other files for the "Getting started with AI on LUMI" workshop☆43Updated 2 months ago
- ALCF Computational Performance Workshop☆38Updated 3 years ago
- The OGB-LSC is the Large Scale Competition by Open Graph Benchmark to help accelerate research into machine learning on graph structured …☆78Updated last year
- Graphium: Scaling molecular GNNs to infinity.☆237Updated 7 months ago
- Sparsity support for PyTorch☆37Updated 8 months ago
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆13Updated 2 years ago
- ☆25Updated 7 months ago
- Graph neural networks in JAX.☆68Updated last year
- SC23 Deep Learning at Scale Tutorial Material☆48Updated last year