Data generation

[puria-radmard]

Generate with teacher model, saving:

• Token string
• Probability over each token
• Early exit probs for each layer on each token

[puria-radmard]

Update schema + sparsify matrices