rlightning

RLightning: Large-scale distributed reinforcement learning framework.

Subpackages

• rlightning.buffer
  • rlightning.buffer.utils
    • rlightning.buffer.utils.postprocess
    • rlightning.buffer.utils.preprocess
    • rlightning.buffer.utils.preprocessors
    • rlightning.buffer.utils.storage
    • rlightning.buffer.utils.table
    • rlightning.buffer.utils.utils
  • rlightning.buffer.base_buffer
  • rlightning.buffer.replay_buffer
  • rlightning.buffer.reservoir_buffer
  • rlightning.buffer.rollout_buffer
  • rlightning.buffer.sampler
  • rlightning.buffer.simple_buffer
• rlightning.engine
  • rlightning.engine.async_rl_engine
  • rlightning.engine.async_rsl_rl_engine
  • rlightning.engine.base_engine
  • rlightning.engine.evaluation_engine
  • rlightning.engine.nfsp_engine
  • rlightning.engine.rsl_rl_engine
  • rlightning.engine.sync_rl_engine
• rlightning.env
  • rlightning.env.remote_env
    • rlightning.env.remote_env.env_client
  • rlightning.env.ale_env
  • rlightning.env.base_env
  • rlightning.env.env_group
  • rlightning.env.env_server
  • rlightning.env.isaac_env
  • rlightning.env.maniskill_env
  • rlightning.env.mujoco_env
  • rlightning.env.piper_env
• rlightning.policy
  • rlightning.policy.rsl_rl_policy
    • rlightning.policy.rsl_rl_policy.ppo_vtrace
    • rlightning.policy.rsl_rl_policy.rsl_rl_policy
  • rlightning.policy.simple_ppo_policy
    • rlightning.policy.simple_ppo_policy.ppo_policy
  • rlightning.policy.vla_policy
    • rlightning.policy.vla_policy.ppo_policy
  • rlightning.policy.base_policy
  • rlightning.policy.policy_group
  • rlightning.policy.supervised_policy
• rlightning.types
  • rlightning.types.batched_data
  • rlightning.types.env_rets
  • rlightning.types.metadata
  • rlightning.types.policy_response
• rlightning.weights
  • rlightning.weights.utils
  • rlightning.weights.weight_buffer
  • rlightning.weights.weight_buffer_mixin
  • rlightning.weights.weight_transfer_manager