terrence-ou / Generative-Deep-Learning-2nd-Edition-PyTorch-JAX
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:
- 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…☆726Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆33Updated 2 months ago
- Notebooks for the "JAX in Action" book☆147Updated 10 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆129Updated last year
- NUS CS5242 Neural Networks and Deep Learning, Xavier Bresson, 2025☆380Updated 3 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated last month
- The Forward-Forward Algorithm for Drug Discovery☆34Updated 2 years ago
- The official repository for the book "Machine Learning for Drug Discovery" (Manning Publications)☆38Updated last month
- Practical machine learning for physicists☆17Updated 2 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆81Updated 11 months ago
- Quantum Computing and Machine Learning for Drug Design and Proteins Engineering☆14Updated last year
- Interactive visualizations of the theory behind diffusion models.☆373Updated this week
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆191Updated this week
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆155Updated last year
- PySCF on IPU☆42Updated last year
- ☆18Updated last year
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆427Updated 6 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆275Updated 11 months ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆131Updated 3 years ago
- Advanced Topics in Artificial Intelligence, NUS CS6208, 2023☆324Updated last year
- The files used for Series: Quantum chemistry calculations with python☆19Updated last year
- Hands-on AlphaFold implementation for educational purposes.☆66Updated last month
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆214Updated this week
- All about the fundamentals and working of Diffusion Models☆155Updated 2 years ago
- ☆49Updated last year
- GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine l…☆101Updated last year
- ☆43Updated 11 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆82Updated 11 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆37Updated last year
- Repository of Quantum Datasets Publicly Available☆47Updated last month