using the new fauxjax api for new