feat: add a Dockerfile

[HarikrishnanBalagopal]

$ docker run --rm -it -v "$PWD/workdir/input.json:/app/input.json" icr.io/ftplatform/fm_training_estimator:latest
INFO:root:Hybrid Estimator: Initializing
config.json: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 662/662 [00:00<00:00, 1.59MB/s]
tokenizer_config.json: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 4.53k/4.53k [00:00<00:00, 10.8MB/s]
tokenizer.json: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 2.06M/2.06M [00:00<00:00, 25.4MB/s]
special_tokens_map.json: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1.02k/1.02k [00:00<00:00, 2.83MB/s]
{
    "total_mem_estimate_og": 787623051264.0,
    "activation_memory_og": 596732018688.0,
    "gradient_memory_og": 47722758144.0,
    "model_memory_og": 47722758144.0,
    "optimizer_memory_og": 95445516288.0,
    "total_mem_estimate": "733.5 GiB",
    "activation_memory": "555.8 GiB",
    "gradient_memory": "44.4 GiB",
    "model_memory": "44.4 GiB",
    "optimizer_memory": "88.9 GiB",
    "tps": 2480.99560546875
}