# MSA TRANSFORMER in jax!
**WARNING** This code is work-in-progress!

<a href="https://colab.research.google.com/github/sokrypton/ColabDesign/blob/v1.1.1/esm_msa/example.ipynb">
  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
</a>

### Contributors:
- Shihao Feng [@JeffSHF](https://github.com/JeffSHF)
- Lin Xu
- Facebook Research - [original Pytorch code](https://github.com/facebookresearch/esm)