Studying the big bang with artificial intelligence