Imagen

Recreating google's imagen using the Jax framework

Name		Name	Last commit message	Last commit date
Latest commit History 1,239 Commits
misc		misc
scripts		scripts
.gitignore		.gitignore
README.md		README.md
T5Utils.py		T5Utils.py
TPUManager.py		TPUManager.py
Trainer.py		Trainer.py
config.py		config.py
config.yaml		config.yaml
createTPU-POD.py		createTPU-POD.py
dataCollector.py		dataCollector.py
dataset_utils.py		dataset_utils.py
driveDataLoaderParallel.py		driveDataLoaderParallel.py
driveDataset.py		driveDataset.py
imagen_main.py		imagen_main.py
layers.py		layers.py
main.py		main.py
model_utils.py		model_utils.py
modeling_imagen.py		modeling_imagen.py
partitioning.py		partitioning.py
processDataset.py		processDataset.py
ray_tpu.py		ray_tpu.py
sampler.py		sampler.py
serve.py		serve.py
startTPU.py		startTPU.py
startpod.sh		startpod.sh
test.py		test.py
uploader.txt		uploader.txt
utils.py		utils.py

Provide feedback