View Source
Rein
(rein v0.1.0)
Reinforcement Learning training and inference framework
@type t() :: %Rein{
agent: module(),
agent_state: term(),
environment: module(),
environment_state: term(),
episode: Nx.t(),
iteration: Nx.t(),
random_key: Nx.t(),
trajectory: Nx.t()
}
@spec train(
{environment :: module(), init_opts :: keyword()},
{agent :: module(), init_opts :: keyword()},
epoch_completed_callback :: (map() -> :ok),
state_to_trajectory_fn :: (t() -> Nx.t()),
opts :: keyword()
) :: term()