If you are looking for generative part it isn’t opensource but you can compile a form to request the model:

  
      

      github.com
  

  
    google-research/ghum at master · google-research/google-research - Model Download 5

  master/ghum

  Google Research. Contribute to google-research/google-research development by creating an account on GitHub.