terrence-ou / Generative-Deep-Learning-2nd-Edition-PyTorch-JAXLinks
Pytorch/JAX implementation of examples in Generative Deep Learning 2nd Edition by David Foster
☆23Updated last year
Alternatives and similar repositories for Generative-Deep-Learning-2nd-Edition-PyTorch-JAX
Users that are interested in Generative-Deep-Learning-2nd-Edition-PyTorch-JAX are comparing it to the libraries listed below
Sorting:
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆52Updated 2 weeks ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆133Updated last year
- ☆36Updated 3 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆38Updated 6 months ago
- ☆32Updated 3 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆84Updated last year
- PySCF on IPU☆44Updated last year
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆181Updated 2 years ago
- TLDRs for ML in Drug Discovery papers☆71Updated 2 years ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆301Updated this week
- The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series o…☆754Updated last year
- Hands-on AlphaFold implementation for educational purposes.☆88Updated 5 months ago
- Advanced Topics in Artificial Intelligence, NUS CS6208, 2023☆327Updated 2 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆108Updated 5 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆85Updated last year
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆51Updated 6 months ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆121Updated 2 months ago
- NUS CS5242 Neural Networks and Deep Learning, Xavier Bresson, 2025☆400Updated 5 months ago
- This repository contains the implementation of the Multi-view Molecular Embedding with Late Fusion (MMELON) architecture. MMELON combines…☆39Updated 3 months ago
- ☆45Updated last year
- Minimal AlphaFold☆54Updated last year
- The official repository for the book "Machine Learning for Drug Discovery" (Manning Publications)☆46Updated 2 weeks ago
- Mixed continous/categorical flow-matching model for de novo molecule generation.☆159Updated 3 weeks ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆55Updated last week
- MESS: Modern Electronic Structure Simulations☆37Updated 3 weeks ago
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆255Updated 3 weeks ago
- jax library for E3 Equivariant Neural Networks☆216Updated last month
- ☆39Updated last year
- [ICML 2025] Repurposing pre-trained score-based generative models for transition path sampling by minimizing the Onsager-Machlup (OM) act…☆17Updated 2 months ago
- The Forward-Forward Algorithm for Drug Discovery☆34Updated 2 years ago