I haven't used ml-agents yet, but it's open source (Apache 2 license). I think the best starting point would be this importer for demo files.