/cloob-training

CLOOB training (JAX) and inference (JAX and PyTorch)

Primary LanguagePythonMIT LicenseMIT

Watchers