flax

514dba8d - Fix bug in graph overhead benchmark