Beta

Explore
Marketplace
Neural Labs
Playground
Wallet
Docs

Beta

Explore Marketplace Neural Labs Wallet Docs

Build

Manage

My assets Manage wallet Organization

Community

Discord X (Twitter)Telegram

Docs Privacy Policy Terms and Conditions

Proud member of

Proud member of

Docs Cookie Policy

Privacy Policy Terms and Conditions

© 2026 AlphaNeural AI. All rights reserved.

t5-vae-python – AI Model by flax-community | AlphaNeural AI

You can deploy this model and start earning money today!

flax-community/
t5-vae-python

like

0

transformer_vae

Fraser/python-lines

endpoints_compatible

Views

No views yet

T5-VAE-Python (flax)

A Transformer-VAE made using flax.

Try the demo!

It has been trained to interpolate on lines of Python code from the python-lines dataset.

Done as part of Huggingface community training (see forum post).

Builds on T5, using an autoencoder to convert it into an MMD-VAE (more info).

How to use from the 🤗/transformers library

Add model repo as a submodule:

git submodule add https://github.com/Fraser-Greenlee/t5-vae-flax.git t5_vae_flax

python

from transformers import AutoTokenizer
from t5_vae_flax.src.t5_vae import FlaxT5VaeForAutoencoding

tokenizer = AutoTokenizer.from_pretrained("t5-base")

model = FlaxT5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-python")

Setup

Run setup_tpu_vm_venv.sh to setup a virtual enviroment on a TPU VM for training.