DavidZhang73 / pytorch-lightning-templateLinks
A template for simple deep learning projects using Lightning
☆24Updated 10 months 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.☆170Updated 2 years ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆276Updated 10 months ago
- A simple cross attention that updates both the source and target in one step☆182Updated 2 months ago
- [ICLR 2025 Spotlight] Boltzmann-Aligned Inverse Folding Model as a Predictor of Mutational Effects on Protein-Protein Interactions☆40Updated 7 months ago
- ☆12Updated 6 months ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆71Updated last year
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆93Updated 2 years ago
- [ICML 2024] Structure-Aware E(3)-Invariant Molecular Conformer Aggregation Networks.☆25Updated 7 months ago
- Code implementation for the paper "Large-scale Pre-training for Grounded Video Caption Generation" (ICCV 2025)☆25Updated last month
- Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration wit…☆308Updated last month
- Implementation of AlphaGenome, Deepmind's updated genomic attention model☆69Updated 2 months ago
- Annotated Flow Matching paper☆213Updated last year
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆67Updated 6 months ago
- Unofficial PyTorch implementation of Discrete Denoising Diffusion Probabilistic Model(D3PM)☆40Updated 2 years ago
- [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"☆60Updated last month
- Simple Guidance Mechanisms for Discrete Diffusion Models☆55Updated 10 months ago
- Code for the paper DisCo-Diff: Enhancing Continuous Diffusion Models with Discrete Latents, ICML 2024☆89Updated last year
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆80Updated 5 months ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆226Updated last year
- ☆48Updated last year
- [ICLR 2024] "Latent 3D Graph Diffusion" by Yuning You, Ruida Zhou, Jiwoong Park, Haotian Xu, Chao Tian, Zhangyang Wang, Yang Shen☆52Updated last year
- [ICLR 2024 Spotlight] The official repo for the paper "De novo Protein Design using Geometric Vector Field Networks".☆30Updated last year
- ☆125Updated last year
- Batch computation of the linear assignment problem on GPU.☆94Updated last month
- ☆79Updated 3 months ago
- Aligned Diffusion Schroedinger Bridges (UAI 2023)☆33Updated 2 years ago
- MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generation☆107Updated last year
- Official Implementation (Pytorch) of the "LLaMo: Large Language Model-based Molecular Graph Assistant", NeurIPS 2024☆32Updated 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…☆241Updated 3 months ago
- Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences li…☆149Updated last year