idejie / PyTorch-Hydra-Template

Related projects ⓘ

Alternatives and complementary repositories for PyTorch-Hydra-Template

• CarlaSa / diffusion_journal_club_24
  Repository of the journal club "Diffusion Models and Generative Modeling"
  ☆14Updated last week
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆227Updated last year
• ebekkers / ponita
  Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.
  ☆73Updated 5 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆49Updated 8 months ago
• e3nn / e3nn-jax
  jax library for E3 Equivariant Neural Networks
  ☆183Updated last month
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆57Updated 8 months ago
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆20Updated last year
• oxcsml / riemannian-score-sde
  Score-based generative models for compact manifolds
  ☆103Updated 10 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆123Updated 10 months ago
• QUVA-Lab / escnn
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆359Updated 2 weeks ago
• klieret / wandb-offline-sync-hook
  A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
  ☆55Updated last week
• Rose-STL-Lab / Approximately-Equivariant-Nets
  ☆18Updated 7 months ago
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆39Updated 9 months ago
• francois-rozet / diffusion-priors
  Learning Diffusion Priors from Observations by Expectation Maximization
  ☆18Updated 2 months ago
• malbergo / stochastic-interpolants
  ☆104Updated 9 months ago
• seongsukim-ml / GPWNO
  Gaussain plane-wave neural operator (GPWNO) is a novel approach to predict the electron density of molecule, combining two types of the b…
  ☆18Updated last month
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆29Updated last year
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆177Updated 8 months ago
• Tikquuss / GflowNets_Tutorial
  GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…
  ☆82Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆199Updated last year
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆74Updated 3 months ago
• QUVA-Lab / e2cnn_experiments
  Experiment for General E(2)-Equivariant Steerable CNNs
  ☆26Updated 3 years ago
• lollcat / se3-augmented-coupling-flows
  SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.
  ☆23Updated 7 months ago
• RobDHess / Steerable-E3-GNN
  E(3) Steerable Graph Neural Network
  ☆110Updated last year
• google-research / e3x
  E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
  ☆89Updated 3 weeks ago
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆255Updated last year
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆73Updated 2 years ago
• fabian-sp / MoMo
  MoMo: Momentum Models for Adaptive Learning Rates
  ☆16Updated 5 months ago
• DreamFold / FoldFlow
  FoldFlow: SE(3)-Stochastic Flow Matching for Protein Backbone Generation
  ☆173Updated 6 months ago
• TianrongChen / DMSB
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆20Updated last year