flax

da8ef361 - Add flip doc link to eager sharding error message