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