tiny-stories-96-seq-len-hpo-run-from-305

[david-thrower]

From #305, integrating some work from #306

Key features: Hyperparamter optimization run from #305 plus

• Increase context length to 96
• Subset of Tinystories 96 context length 50k data set