Default to mkdirs behavior in async_makedirs.

PiperOrigin-RevId: 464131053

orbax/checkpoint/utils.py

@@ -47,8 +47,12 @@ async def run(*args, loop=None, executor=None, **kwargs):
 
 
 # TODO(cpgaffney): This functionality should be provided by an external library.
-def async_makedirs(path, *args, **kwargs):
-  return _wrap(path.mkdir)(*args, **kwargs)
+def async_makedirs(path,
+                   *args,
+                   parents: bool = True,
+                   exist_ok: bool = True,
+                   **kwargs):
+  return _wrap(path.mkdir)(*args, parents=parents, exist_ok=exist_ok, **kwargs)
 
 
 def register_ts_spec_for_serialization():