Implementation and replication of ProGen, Language Modeling for Protein Generation, in Jax - View it on GitHub