kitae# kitae package Subpackages kitae.algos package Subpackages Submodules kitae.algos.experience module kitae.algos.factory module Module contents kitae.envs package Subpackages Submodules kitae.envs.make module Module contents kitae.loops package Submodules kitae.loops.train module kitae.loops.update module Module contents kitae.modules package Submodules kitae.modules.encoder module kitae.modules.modules module kitae.modules.optimizer module kitae.modules.policy module kitae.modules.pytree module kitae.modules.qvalue module kitae.modules.value module Module contents kitae.operations package Submodules kitae.operations.loss module kitae.operations.timesteps module kitae.operations.transformation module Module contents Submodules kitae.actor module PolicyActor PolicyActor.select_action() kitae.agent module AgentInfo AgentInfo.config AgentInfo.extra AgentSerializable AgentSerializable.serialize() AgentSerializable.unserialize() BaseAgent BaseAgent.explore() BaseAgent.interact_keys() BaseAgent.restore() BaseAgent.resume() BaseAgent.select_action() BaseAgent.serializable_dict BaseAgent.train() BaseAgent.unserialize() BaseAgent.update() OffPolicyAgent OffPolicyAgent.algo_type OffPolicyAgent.should_update() OffPolicyAgent.step OnPolicyAgent OnPolicyAgent.algo_type OnPolicyAgent.should_update() kitae.buffer module Buffer Buffer.rng Buffer.max_buffer_size Buffer.buffer Buffer.add() Experience Experience.action Experience.done Experience.log_prob Experience.next_observation Experience.observation Experience.reward OffPolicyBuffer OffPolicyBuffer.sample() OnPolicyBuffer OnPolicyBuffer.sample() batchify() batchify_and_randomize() buffer_factory() jax_stack_experiences() numpy_stack_experiences() kitae.config module AlgoConfig AlgoConfig.algo_params AlgoConfig.env_cfg AlgoConfig.seed AlgoConfig.train_cfg AlgoConfig.update_cfg AlgoParams ConfigSerializable ConfigSerializable.serialize() ConfigSerializable.unserialize() EnvConfig EnvConfig.action_space EnvConfig.n_agents EnvConfig.n_envs EnvConfig.observation_space EnvConfig.task_name TrainConfig TrainConfig.n_env_steps TrainConfig.save_frequency UpdateConfig UpdateConfig.batch_size UpdateConfig.learning_rate UpdateConfig.learning_rate_annealing UpdateConfig.max_buffer_size UpdateConfig.max_grad_norm UpdateConfig.n_epochs UpdateConfig.shared_encoder kitae.interface module AlgoType AlgoType.OFF_POLICY AlgoType.ON_POLICY IActor IActor.select_action() IAgent IAgent.explore() IAgent.resume() IAgent.select_action() IAgent.should_update() IAgent.train() IBuffer IBuffer.add() IBuffer.sample() kitae.saving module SaverContext SaverContext.saver SaverContext.update() default_run_name() kitae.types module kitae.version module Module contents
