What's next? Next word prediction with PyTorch
Today, I want to take you through a simple next-word prediction model built using PyTorch. These models are famous in search applications, Gmail’s Smart Compose feature, which predicts the next words or sentences in emails you want to write. Google goes into detail on how they build their smart compose feature in their research blog post here. From this, I want to pull out some key learnings and requirements: Latency. Latency is important, must generate a response in under 100ms....