terrence-ou / Generative-Deep-Learning-2nd-Edition-PyTorch-JAXLinks
Pytorch/JAX implementation of examples in Generative Deep Learning 2nd Edition by David Foster
☆23Updated 2 years ago
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:
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆136Updated last year
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆186Updated 2 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆84Updated last year
- 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…☆760Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆39Updated 7 months ago
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆56Updated last week
- NUS CS5242 Neural Networks and Deep Learning, Xavier Bresson, 2025☆401Updated 6 months ago
- ☆37Updated 4 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆109Updated 6 months ago
- The official repository for the book "Machine Learning for Drug Discovery" (Manning Publications)☆46Updated this week
- Advanced Topics in Artificial Intelligence, NUS CS6208, 2023☆327Updated 2 years ago
- ☆32Updated 3 years ago
- TLDRs for ML in Drug Discovery papers☆71Updated 2 years ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆121Updated 2 months ago
- Hands-on AlphaFold implementation for educational purposes.☆91Updated 6 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆56Updated last month
- A collection of awesome mathematics and computer science courses☆132Updated 9 months ago
- Deep Learning, an Energy Approach☆215Updated 4 months ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆309Updated this week
- Higher order equivariant graph neural networks for 3D point clouds☆44Updated 2 years ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Notebooks for the "Deep Learning with JAX" book☆156Updated 4 months ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆63Updated 6 months ago
- The Forward-Forward Algorithm for Drug Discovery☆34Updated 2 years ago
- Code repository of ml-without-tears blog: https://mlwithouttears.com/☆36Updated last year
- jax library for E3 Equivariant Neural Networks☆217Updated last month
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆86Updated last year
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆263Updated last month
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆52Updated 6 months ago
- Deep Learning Fundamentals -- Code material and exercises☆402Updated last year