Path	Lines of Code
grok/__init__.py	5
grok/data.py	296
grok/measure.py	82
grok/metrics.py	141
grok/training.py	669
grok/transformer.py	292
grok/visualization.py	409
nbs/flatness.ipynb	71
scripts/compute_sharpness.py	10
scripts/create_metric_graphs.py	218
scripts/create_metrics_for_epochs.py	59
scripts/create_partial_metrics.py	113
scripts/make_data.py	6
scripts/train.py	9
scripts/visualize_metrics.py	439
setup.py	16