Small Language Model Experiment building a small language model using TinyStories dataset with PyTorch and MLX