Vae Python Packages

nnm

Neural Network Models

mlx-taef

Tiny AutoEncoders (TAESD family) for diffusion latents on Apple Silicon — pure MLX. Live previews + low-memory decode for FLUX.1, FLUX.2 Klein, SD1.x, SDXL.

pythae

Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)

numpy-ml

Machine learning, in numpy

avitai-artifex

A research-focused modular generative modeling library built on JAX/Flax NNX

disent

🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib

molgen

Lightweight toolkit for de novo molecular generation: SMILES & SELFIES tokenizers, CharRNN / MolGPT / VAE models, training, sampling, and MOSES-style metrics.

human-body-prior

VPoser: Variational Human Pose Prior

ciss-vae

Python package for CISS-VAE.

pi-vae-pytorch

Implementation of Poisson Identifiable VAE (pi-VAE) in Pytorch

easyvae

Python implementation of a Classical and Variational Autoencoders using Numpy

moai-mdk

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

isograph

A Python framework for discovering isoform-switch and splicing modules from bulk RNA-seq by combining gene-local compositional modeling with splice-graph-aware latent network inference.

bucketsampler

Aspect ratio bucketing toolkit for diffusion model training. PyTorch native, DDP correct.

cadl

Python package with source code from the course "Creative Applications of Deep Learning w/ TensorFlow"

hyperion-ml

Python toolkit for speech processing

pytorch-policy

This repository contains model-free deep reinforcement learning algorithms implemented in Pytorch

tsgm

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

pyroved

Variational encoder-decoder models in Pyro probabilistic programming language

daisyrec

An easy-to-use library for recommender systems.

rapidae

Rapidae: Python Library for Rapid Creation and Experimentation of Autoencoders

rectokens

A tokenizer library for sequential recommendation systems

flexinet

Flexible torch neural network architecture API

omniinfer-client

Python SDK for Stable Diffusion API (Txt2Img/Img2Img/ControlNet/VAE)