Add small graph task

[bengioe]

I've been using this environment to "debug" gflownets. PR not ready yet but this branch may be useful for others.