flax

76bb33ab - Add main.py as the entry point for running the ogbg_molpcba example.