Expose DuellingMLP in jax networks package.
@@ -34,6 +34,7 @@ from acme.jax.networks.distributional import CategoricalValueHead
from acme.jax.networks.distributional import GaussianMixture
from acme.jax.networks.distributional import MultivariateNormalDiagHead
from acme.jax.networks.distributional import NormalTanhDistribution
+from acme.jax.networks.duelling import DuellingMLP
from acme.jax.networks.multiplexers import CriticMultiplexer
from acme.jax.networks.policy_value import PolicyValueHead
from acme.jax.networks.rescaling import ClipToSpec
