This repository provides a flexible wrapper around TensorFlow’s tensor2tensor library to train, interact with, and generate data for neural chatbot systems. It integrates multiple dialogue datasets and supports any model or hyperparameter configuration available in tensor2tensor. The project includes a base class for dialog problems that can be easily extended for custom chatbot research and experimentation.