saforem2 / ezpzLinks
Train across all your devices, ezpz 🍋
☆24Updated last 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☆25Updated this week
- ☆59Updated 2 years ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆63Updated last week
- Running Jax in PyTorch Lightning☆113Updated 9 months ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆36Updated 4 years ago
- SciML Benchmarking Suite for AI for Science☆42Updated last year
- Discovering Interpretable Features in Protein Language Models via Sparse Autoencoders☆193Updated this week
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆133Updated last year
- Distributed PyTorch implementation of multi-headed graph convolutional neural networks☆88Updated this week
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆101Updated 2 months ago
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆19Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆16Updated last week
- GenSLMs: Genome-scale language models reveal SARS-CoV-2 evolutionary dynamics☆136Updated last year
- Residue Level Alignment☆19Updated 9 months ago
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆22Updated 5 months ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆274Updated 9 months ago
- Graphium: Scaling molecular GNNs to infinity.☆235Updated 4 months ago
- Minimal AlphaFold☆53Updated last year
- RITA is a family of autoregressive protein models, developed by LightOn in collaboration with the OATML group at Oxford and the Debora Ma…☆99Updated 2 years ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆18Updated 8 months ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆299Updated this week
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated last year
- [ICLR 2022] Data-Efficient Graph Grammar Learning for Molecular Generation☆96Updated 2 years ago
- A parallel framework for training deep neural networks☆63Updated 6 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
- FLOPS counter for all your GPU benchmarking needs☆13Updated last year
- Graph neural networks in JAX.☆67Updated last year
- Use Jax functions in Pytorch☆249Updated 2 years ago
- Sparsity support for PyTorch☆37Updated 5 months ago