GitHub repository: ryanccarelli/ssljax
Primary language: Python
Description: self-supervised learning in jax