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:
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆131Updated last year
- NUS CS5242 Neural Networks and Deep Learning, Xavier Bresson, 2025☆395Updated 3 months ago
- Advanced Topics in Artificial Intelligence, NUS CS6208, 2023☆323Updated 2 years ago
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆41Updated 2 weeks ago
- The official repository for the book "Machine Learning for Drug Discovery" (Manning Publications)☆44Updated 3 months ago
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆168Updated last year
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆84Updated last year
- Official Github for Wharton STAT 4830☆40Updated last month
- TLDRs for ML in Drug Discovery papers☆71Updated 2 years ago
- A collection of awesome mathematics and computer science courses☆127Updated 6 months ago
- 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…☆746Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆36Updated 4 months ago
- ☆50Updated last year
- PySCF on IPU☆42Updated last year
- Programming for AI Practice☆13Updated 4 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆85Updated last year
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆109Updated 2 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆279Updated last year
- Graph Machine Learning course, Xavier Bresson, 2023☆613Updated 10 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…☆264Updated this week
- The Forward-Forward Algorithm for Drug Discovery☆34Updated 2 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆109Updated 3 months ago
- ☆44Updated last year
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆132Updated 3 years ago
- Course notes☆703Updated last year
- Minimal package to demonstrate good Python development habits☆104Updated 3 months ago
- torch-molecule is a deep learning package for molecular discovery, designed with an sklearn-style interface for property prediction, inve…☆265Updated last week
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆242Updated last month
- 🍓 Build and train energy-based and diffusion models in PyTorch ⚡.☆20Updated 2 months ago
- ☆37Updated 11 months ago