flax

DOC: removed optimizer.compute_gradient from guided tour

#167

Merged