Loosen tolerance in jax tests#51

Olllom wants to merge 5 commits into

mainnoegroup/bgflow:mainfrom

loosen-jax-tolnoegroup/bgflow:loosen-jax-tolCopy head branch name to clipboard