add support for bert training to generate_sharktank and pytest+benchmark

[dan-garvey]

This has a lot of changes to the infra for generating tank and running benchmarks for training workload, so please review thoroughly.

As we add more support for training models, I imagine there can be some improvements to remove some boilerplate I've added here.