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
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆64Updated 2 weeks ago
- ☆61Updated 2 years ago
- Proxy server to Argo API, OpenAI format compatible☆15Updated 2 months ago
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆23Updated 7 months ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- SciML Benchmarking Suite for AI for Science☆42Updated last year
- Parallel framework for training and fine-tuning deep neural networks☆65Updated last week
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆92Updated last week
- 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
- Running Jax in PyTorch Lightning☆113Updated 10 months ago
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆37Updated 4 years ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆18Updated 9 months ago
- AI Training Series Material☆38Updated 3 weeks ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆136Updated last year
- Sparsity support for PyTorch☆37Updated 7 months ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated last year
- Collection of small examples for running on ALCF resources☆19Updated 3 months ago
- [ICML 2024 Oral] LSH-Based Efficient Point Transformer (HEPT)☆22Updated 9 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆104Updated 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 …☆79Updated last year
- ☆47Updated 3 months ago
- Residue Level Alignment☆19Updated 11 months ago
- Cosmic Tagging Network for Neutrino Physics☆13Updated last year
- ESM2 protein language models in JAX/Flax☆17Updated 3 years ago
- Repository of machine learning benchmarks☆43Updated last week
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated 2 years ago
- 🍓 Build and train energy-based and diffusion models in PyTorch ⚡.☆54Updated last week
- Library for steering campaigns of simulations on supercomputers☆58Updated 4 months ago