Utilities optimizer.py Contains an implementation of the ADAM optimizer parse_args.py Contains the different flags used by the model viz.py Some functions for visualization