import numpy as np from practice_check import report from support import TEXT, VOCAB, decode, encode def train(ids, vocab_size, rng): raise NotImplementedError def next_log_probs(model, prev_ids): raise NotImplementedError