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☆189Updated 2 years ago
- NUS CS5242 Neural Networks and Deep Learning, Xavier Bresson, 2025☆402Updated 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…☆762Updated last year
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆63Updated last month
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆85Updated last year
- Advanced Topics in Artificial Intelligence, NUS CS6208, 2023☆327Updated 2 years ago
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆221Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆39Updated 8 months ago
- ☆54Updated last year
- TLDRs for ML in Drug Discovery papers☆71Updated 2 years ago
- ☆39Updated 4 months ago
- The official repository for the book "Machine Learning for Drug Discovery" (Manning Publications)☆50Updated 2 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆109Updated 7 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆31Updated 2 years ago
- ☆34Updated 4 months ago
- Official Github for Wharton STAT 4830☆44Updated 4 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆293Updated last year
- Notebooks for the "Deep Learning with JAX" book☆158Updated 5 months ago
- Materials of the Nordic Probabilistic AI School 2022.☆179Updated 3 years ago
- Deep Learning, an Energy Approach☆218Updated 5 months ago
- PySCF on IPU☆44Updated last year
- Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks (ICML …☆508Updated last month
- jax library for E3 Equivariant Neural Networks☆217Updated 2 months ago
- Generative Flow Networks - GFlowNet☆292Updated this week
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆477Updated last year
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆320Updated 2 weeks ago
- Neat Bayesian machine learning examples☆58Updated last week
- ☆34Updated 3 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆86Updated last year