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 2 weeks ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆64Updated last month
- ☆61Updated 2 years ago
- SciML Benchmarking Suite for AI for Science☆42Updated last year
- Parallel framework for training and fine-tuning deep neural networks☆64Updated 6 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆16Updated last week
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- Running Jax in PyTorch Lightning☆113Updated 9 months ago
- SC23 Deep Learning at Scale Tutorial Material☆47Updated last year
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆13Updated 2 years ago
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆22Updated 6 months ago
- Collection of small examples for running on ALCF resources☆19Updated 2 months ago
- ☆228Updated last month
- Cosmic Tagging Network for Neutrino Physics☆13Updated last year
- Repository of machine learning benchmarks☆42Updated last week
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆135Updated last year
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆308Updated this week
- ☆46Updated 2 months ago
- Einsum-like high-level array sharding API for JAX☆35Updated last year
- Library for steering campaigns of simulations on supercomputers☆58Updated 4 months ago
- A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/☆33Updated 6 months ago
- AI Training Series Material☆37Updated last week
- ☆128Updated 10 months ago
- Graphium: Scaling molecular GNNs to infinity.☆234Updated 5 months ago
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆90Updated this week
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆84Updated last year
- ☆115Updated last month
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆144Updated 5 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
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆101Updated 3 months ago