Function Approximators¶

Collection of useful Neural Networks.

jax_agents.common.networks.mlp_policy_net(x, output_sizes)¶: Return simple feedforward neural network with tanh output.

jax_agents.common.networks.mlp_value_net(x, output_sizes)¶: Return simple feedforward neural network with tanh output.