Anon

Author: yardenas

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 Yarden As
+Copyright (c) 2024 Anon
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

pyproject.toml

@@ -2,7 +2,7 @@
 name = "act-safe"
 version = "0.1.0"
 description = ""
-authors = ["Yarden As"]
+authors = ["Anon"]
 readme = "README.md"
 
 [tool.poetry.dependencies]

safe_opax/common/double_buffer.py

@@ -27,7 +27,7 @@ def double_buffer(ds: Iterable[TrajectoryData]) -> Iterator[TrajectoryData]:
       Batches of sharded device arrays.
     """
     batch = None
-    # TODO (yarden): should be sharded better
+    # TODO (anon): should be sharded better
     # see equinox's docs for DDP
     devices = jax.local_devices()[0]
     for next_batch in ds:

safe_opax/common/learner.py

@@ -27,7 +27,7 @@ def __init__(
     def grad_step(
         self, model: PyTree, grads: PyTree, state: optax.OptState
     ) -> tuple[PyTree, optax.OptState]:
-        # TODO (yarden): should retain the policy and use params dtype
+        # TODO (anon): should retain the policy and use params dtype
         grads = apply_dtype(grads, jnp.float32)
         updates, new_opt_state = self.optimizer.update(grads, state)
         all_ok = all_finite(updates)