[Pytorch] Generative retrieval model based on RQ-VAE from "Recommender Systems with Generative Retrieval"

PyKnot code for Knot Classification & Generation

moai is a PyTorch-based AI Model Development Kit (MDK) created to improve data-driven model workflows, design and reproducibility.

Public repository for Unsupervised Binary Variational Auto-Encoder (BVAE) for Hashing

OmniTokenizer: one model and one weight for image-video joint tokenization.

Python toolkit for speech processing

Generation and evaluation of synthetic time series datasets (also, augmentations, visualizations, a collection of popular datasets)

[WIP] RL agent for the SuperTuxKart game.

This repo combines Proximal Policy Optimization (PPO) with a geometric linear velocity control for path-following and collision avoidance in 3D environments. A Variational Autoencoder (VAE) compresses high-dimensional depth image inputs from the drone's depth camera, the sole exteroceptive sensor. Digital Twins (DT) test the sim2real readiness.

PyTorch Studio

Tinygrad and PubDAS

VAEs with PyTorch + Lightning

Implementation of GAN and VAE on MNIST as well as Fashion MNIST

List of molecular design using Generative AI and Deep Learning

List of protein (enzymes, antibody, and PPIs) conformations and molecular dynamics using generative artificial intelligence and deep learning

Сustom torch style machine learning framework with automatic differentiation implemented on numpy, allows build GANs, VAEs, etc.

A Collection of Variational Autoencoders (VAE) in PyTorch.

This project uses a Variational Autoencoder (VAE) to generate SMILES strings for novel compound generation. The VAE model is trained on a dataset of existing chemical compounds and can generate new, valid SMILES strings, which may represent potentially new and useful chemical entities.

Diffusion Models in Medical Imaging (Published in Medical Image Analysis Journal)