Improve Transformers support (#2970)

Commit

362 days ago

Improve Transformers support (#2970) * Much better support * add gpt neox * bump transformers version * bump version