updated fauxjax to 0.2.1