diff --git a/acme/jax/__init__.py b/acme/jax/__init__.py index 240cb71526..7bbfbb386c 100644 --- a/acme/jax/__init__.py +++ b/acme/jax/__init__.py @@ -12,3 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +"""JAX-specific components for Acme.""" + +# Expose the experiments submodule. +from acme.jax import experiments