DavidZhang73 / pytorch-lightning-templateLinks
A template for simple deep learning projects using Lightning
☆26Updated 3 weeks ago
Alternatives and similar repositories for pytorch-lightning-template
Users that are interested in pytorch-lightning-template are comparing it to the libraries listed below
Sorting:
- A simple python package to solve all your problems with pythonpath, work dir, file paths, module imports and environment variables.☆173Updated 2 years ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆283Updated last year
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆73Updated 2 years ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆88Updated last week
- Code for “FlowMM Generating Materials with Riemannian Flow Matching” and "FlowLLM: Flow Matching for Material Generation with Large Langu…☆170Updated last year
- Python package for P2 (Path Planning), a masked diffusion model sampling method for sequence generation (protein, text, etc.).☆20Updated 4 months ago
- [ICLR 2024] "Latent 3D Graph Diffusion" by Yuning You, Ruida Zhou, Jiwoong Park, Haotian Xu, Chao Tian, Zhangyang Wang, Yang Shen☆53Updated last year
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆68Updated 8 months ago
- [ICLR 2022] The implementation for the paper "Equivariant Graph Mechanics Networks with Constraints".☆63Updated last year
- [ICLR 2025 Spotlight] Boltzmann-Aligned Inverse Folding Model as a Predictor of Mutational Effects on Protein-Protein Interactions☆42Updated 9 months ago
- MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generation☆109Updated last year
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆98Updated 2 years ago
- ☆12Updated 8 months ago
- ☆49Updated 2 years ago
- ☆126Updated last year
- ☆21Updated 9 months ago
- ☆34Updated 10 months ago
- ☆51Updated last year
- Implementation of paper "Uni-3DAR: Unified 3D Generation and Understanding via Autoregression on Compressed Spatial Tokens"☆138Updated 8 months ago
- Multi-modal Molecule Structure-text Model for Text-based Editing and Retrieval, Nat Mach Intell 2023 (https://www.nature.com/articles/s42…☆249Updated 5 months ago
- Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration wit…☆320Updated 3 months ago
- [ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs☆265Updated 10 months ago
- Implementation of the Paper "Permutation-Invariant Variational Autoencoder for Graph-Level Representation Learning" by Robin Winter, Fran…☆44Updated 2 years ago
- [ICLR 2024 Spotlight] The official repo for the paper "De novo Protein Design using Geometric Vector Field Networks".☆30Updated last year
- Annotated Flow Matching paper☆222Updated last year
- PRESTO: Progressive Pretraining Enhances Synthetic Chemistry Outcomes [EMNLP 2024]☆28Updated last year
- [ICML 2024] Structure-Aware E(3)-Invariant Molecular Conformer Aggregation Networks.☆24Updated 9 months ago
- SwinGNN: diffusion model for graph generation☆86Updated last year
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆226Updated last year
- Aligned Diffusion Schroedinger Bridges (UAI 2023)☆33Updated 2 years ago