Vae Python Packages

nnm

Neural Network Models

pythae

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

avitai-artifex

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

isograph

Python toolkit for discovering isoform-switch and splicing modules from bulk RNA-seq using gene-aware latent network inference.

numpy-ml

Machine learning, in numpy

ciss-vae

Python package for CISS-VAE.

human-body-prior

VPoser: Variational Human Pose Prior

bucketsampler

Aspect ratio bucketing for diffusion training. PyTorch native, DDP correct, no training-framework lock-in.

disent

Vae disentanglement framework built with pytorch lightning.

moai-mdk

moai: Accelerated, Flexible, Modular, Reproducible, Insightful AI

pi-vae-pytorch

A Pytorch implementation of Poisson Identifiable VAE (pi-VAE), a variational auto encoder used to construct latent variable models of neural activity while simultaneously modeling the relation between the latent and task variables.

cadl

Creative Applications of Deep Learning with TensorFlow

easyvae

Python implementation of a Classical and Variational Autoencoders using Numpy

pytorch-policy

Reinforcement Learning in Pytorch

tsgm

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

pyroved

Invariant representation learning from imaging and spectral data

hyperion-ml

Python toolkit for speech processing

daisyrec

Official code for "DaisyRec 2.0: Benchmarking Recommendation for Rigorous Evaluation" (TPAMI2022) and "Are We Evaluating Rigorously? Benchmarking Recommendation for Reproducible Evaluation and Fair Comparison" (RecSys2020)

omniinfer-client

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

flexinet

flexinet: flexible torch neural network composition

rapidae

Explore, compare and develop autoencoder models with a back-end agnostic framework

rectokens

[Pytorch] Efficient tokenization library for recommendations and generative retrieval using pre-trained language models. Inspired by "Vectorizing the Trie: Efficient Constrained Decoding for LLM-based Generative Retrieval on Accelerators"

hydra-moai-searchpath-plugin

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

pvae

code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".