DavidZhang73 / pytorch-lightning-templateLinks
A template for simple deep learning projects using Lightning
☆24Updated 8 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.☆169Updated 2 years ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆71Updated last year
- Annotated Flow Matching paper☆205Updated 11 months ago
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆65Updated 5 months ago
- SwinGNN: diffusion model for graph generation☆85Updated last year
- Batch computation of the linear assignment problem on GPU.☆92Updated last month
- ☆32Updated last month
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆273Updated 8 months ago
- ☆48Updated last year
- Code implementation for the paper "Large-scale Pre-training for Grounded Video Caption Generation" (ICCV 2025)☆19Updated this week
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆83Updated last month
- [ICLR 2025 Spotlight] Boltzmann-Aligned Inverse Folding Model as a Predictor of Mutational Effects on Protein-Protein Interactions☆39Updated 5 months ago
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆92Updated 2 years ago
- [ICLR 2022] The implementation for the paper "Equivariant Graph Mechanics Networks with Constraints".☆62Updated 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
- ☆118Updated last year
- ☆12Updated 5 months ago
- Code for the paper DisCo-Diff: Enhancing Continuous Diffusion Models with Discrete Latents, ICML 2024☆86Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆59Updated 6 months ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆50Updated 8 months ago
- Unify and Simplify Discrete-time and Continuous-time Discrete Denoising Diffusion☆23Updated 7 months ago
- A simple cross attention that updates both the source and target in one step☆178Updated last month
- Code for “FlowMM Generating Materials with Riemannian Flow Matching” and "FlowLLM: Flow Matching for Material Generation with Large Langu…☆154Updated 10 months ago
- Implementation of the Paper "Permutation-Invariant Variational Autoencoder for Graph-Level Representation Learning" by Robin Winter, Fran…☆42Updated 2 years ago
- Implementation of paper "Uni-3DAR: Unified 3D Generation and Understanding via Autoregression on Compressed Spatial Tokens"☆128Updated 4 months ago
- This is the official code release for Bayesian Flow Networks.☆293Updated last year
- Unofficial PyTorch implementation of Discrete Denoising Diffusion Probabilistic Model(D3PM)☆39Updated 2 years ago
- ☆19Updated 5 months ago
- Implementation of AlphaGenome, Deepmind's updated genomic attention model☆66Updated last month
- RetroBridge: Markov Bridge Model for Retrosynthesis Planning☆31Updated last year