litgpt
litgpt copied to clipboard
Compute TPU FLOPs