DavidZhang73 / pytorch-lightning-templateLinks
A template for simple deep learning projects using Lightning
☆23Updated 7 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.☆165Updated 2 years ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆273Updated 7 months ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆71Updated last year
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆63Updated 4 months ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆82Updated last week
- Annotated Flow Matching paper☆196Updated 10 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…☆144Updated 10 months ago
- ☆8Updated 3 years ago
- Unify and Simplify Discrete-time and Continuous-time Discrete Denoising Diffusion☆23Updated 6 months ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆225Updated last year
- Unofficial PyTorch implementation of Discrete Denoising Diffusion Probabilistic Model(D3PM)☆39Updated 2 years ago
- ☆32Updated last month
- [ICLR 2024] "Latent 3D Graph Diffusion" by Yuning You, Ruida Zhou, Jiwoong Park, Haotian Xu, Chao Tian, Zhangyang Wang, Yang Shen☆51Updated last year
- Multi-modal Molecule Structure-text Model for Text-based Editing and Retrieval, Nat Mach Intell 2023 (https://www.nature.com/articles/s42…☆234Updated last month
- Simple Guidance Mechanisms for Discrete Diffusion Models☆48Updated 7 months ago
- simple bibtex generator for any text with \cite{}☆31Updated last year
- Aligned Diffusion Schroedinger Bridges (UAI 2023)☆32Updated last year
- Code implementation for the paper "Large-scale Pre-training for Grounded Video Caption Generation" (ICCV 2025)☆19Updated last month
- [ICML 2024] Structure-Aware E(3)-Invariant Molecular Conformer Aggregation Networks.☆25Updated 5 months ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆486Updated last year
- [ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs☆246Updated 6 months ago
- ☆47Updated last year
- [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"☆46Updated this week
- Code for “FlowMM Generating Materials with Riemannian Flow Matching” and "FlowLLM: Flow Matching for Material Generation with Large Langu…☆152Updated 9 months ago
- [ICLR 2022] The implementation for the paper "Equivariant Graph Mechanics Networks with Constraints".☆62Updated last year
- Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration wit…☆296Updated 2 years ago
- A simple cross attention that updates both the source and target in one step☆173Updated last week
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆92Updated 2 years ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆99Updated last month
- Official Jax Implementation of MD4 Masked Diffusion Models☆120Updated 5 months ago