Updated fluidsim example to have a gradient check, and be more general.
Updated fluidsim example to have a gradient check, and be more general.

google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang