CHANGELOG.md
CITATION.cff
LICENSE
MANIFEST.in
README.md
pyproject.toml
docs/README.md
docs/api.md
docs/benchmark_protocol.md
docs/gpu_setup.md
docs/learning.md
docs/recording.md
docs/scenarios.md
docs/assets/find-opening.gif
docs/assets/wall.gif
examples/random_policy.py
examples/record_policy.py
swarmbots/__init__.py
swarmbots.egg-info/PKG-INFO
swarmbots.egg-info/SOURCES.txt
swarmbots.egg-info/dependency_links.txt
swarmbots.egg-info/entry_points.txt
swarmbots.egg-info/requires.txt
swarmbots.egg-info/top_level.txt
swarmbots/benchmark/__init__.py
swarmbots/benchmark/cli.py
swarmbots/benchmark/evaluation.py
swarmbots/benchmark/metadata.py
swarmbots/benchmark/recording.py
swarmbots/benchmark/registry.py
swarmbots/learn/__init__.py
swarmbots/learn/base_policy.py
swarmbots/learn/base_sampler.py
swarmbots/learn/benchmark_policy.py
swarmbots/learn/checkpoint_recording.py
swarmbots/learn/checkpointing.py
swarmbots/learn/discord_notifications.py
swarmbots/learn/evaluation.py
swarmbots/learn/exponential_moving_average.py
swarmbots/learn/gsde_reset.py
swarmbots/learn/hybrid_action_space.py
swarmbots/learn/losses.py
swarmbots/learn/masking.py
swarmbots/learn/metrics_list.py
swarmbots/learn/metrics_logger.py
swarmbots/learn/obs_indices.py
swarmbots/learn/performance_timer.py
swarmbots/learn/polyak_update.py
swarmbots/learn/recording.py
swarmbots/learn/rollout_utils.py
swarmbots/learn/serialization_utils.py
swarmbots/learn/summary_statistics.py
swarmbots/learn/swarmbots_obs_indices.py
swarmbots/learn/temporal_state.py
swarmbots/learn/tensor_conversion.py
swarmbots/learn/testing_env.py
swarmbots/learn/torch_device.py
swarmbots/learn/torch_logging.py
swarmbots/learn/torch_running_mean_std.py
swarmbots/learn/training.py
swarmbots/learn/action_dists/__init__.py
swarmbots/learn/action_dists/action_dist.py
swarmbots/learn/action_dists/action_sampling.py
swarmbots/learn/action_dists/bernoulli_action_dist.py
swarmbots/learn/action_dists/beta_action_dist.py
swarmbots/learn/action_dists/continuous_action_dist.py
swarmbots/learn/action_dists/diag_gaussian_action_dist.py
swarmbots/learn/action_dists/discrete_action_dist.py
swarmbots/learn/action_dists/entropy_utils.py
swarmbots/learn/action_dists/gsde_action_dist.py
swarmbots/learn/action_dists/gumbel_softmax_sign_magnitude_action_dist.py
swarmbots/learn/action_dists/hybrid_action_dist.py
swarmbots/learn/action_dists/predicted_std_gaussian_action_dist.py
swarmbots/learn/action_dists/sign_magnitude_action_dist.py
swarmbots/learn/action_dists/sign_magnitude_beta_action_dist.py
swarmbots/learn/action_dists/squashed_diag_gaussian_action_dist.py
swarmbots/learn/action_dists/tanh_bijector.py
swarmbots/learn/action_dists/temporally_correlated_action_dist.py
swarmbots/learn/algos/__init__.py
swarmbots/learn/algos/base_algorithm.py
swarmbots/learn/algos/mat_qc_base_policy.py
swarmbots/learn/algos/mappo/__init__.py
swarmbots/learn/algos/mappo/mappo_actor.py
swarmbots/learn/algos/mappo/mappo_policy.py
swarmbots/learn/algos/mat/__init__.py
swarmbots/learn/algos/mat/mat_dec_policy.py
swarmbots/learn/algos/mat/mat_encoder.py
swarmbots/learn/algos/mat/mat_ind_policy.py
swarmbots/learn/algos/mat_orig/__init__.py
swarmbots/learn/algos/mat_orig/mat_orig_decoder.py
swarmbots/learn/algos/mat_orig/mat_orig_encoder.py
swarmbots/learn/algos/mat_orig/mat_orig_policy.py
swarmbots/learn/algos/mat_qcx/__init__.py
swarmbots/learn/algos/mat_qcx/mat_qcx_decoder.py
swarmbots/learn/algos/mat_qcx/mat_qcx_policy.py
swarmbots/learn/algos/off_policy/__init__.py
swarmbots/learn/algos/off_policy/off_policy_rollout.py
swarmbots/learn/algos/off_policy/replay_buffer.py
swarmbots/learn/algos/off_policy/replay_buffer_tensor_ops.py
swarmbots/learn/algos/ppo/__init__.py
swarmbots/learn/algos/ppo/base_ppo_policy.py
swarmbots/learn/algos/ppo/ppo.py
swarmbots/learn/algos/ppo/ppo_policy.py
swarmbots/learn/algos/ppo/ppo_rollout.py
swarmbots/learn/algos/ppo/ppo_rollout_batch.py
swarmbots/learn/algos/ppo/ppo_rollout_buffer.py
swarmbots/learn/algos/ppo/ppo_sampler.py
swarmbots/learn/algos/r_mat/__init__.py
swarmbots/learn/algos/r_mat/r_mat_encoder.py
swarmbots/learn/algos/r_mat/r_mat_ind_policy.py
swarmbots/learn/algos/r_mat/r_mat_policy_mixin.py
swarmbots/learn/algos/r_mat/r_mat_qcx_policy.py
swarmbots/learn/algos/r_mat/r_ppo_wm_sampler.py
swarmbots/learn/algos/r_mat/temporal_sequence_model.py
swarmbots/learn/algos/sac/__init__.py
swarmbots/learn/algos/sac/base_sac_policy.py
swarmbots/learn/algos/sac/recurrent_sac.py
swarmbots/learn/algos/sac/recurrent_tmasac_policy.py
swarmbots/learn/algos/sac/sac.py
swarmbots/learn/algos/sac/sac_nop.py
swarmbots/learn/algos/sac/sac_tensor_ops.py
swarmbots/learn/algos/sac/scenario_obs_encoder.py
swarmbots/learn/algos/sac/segment_tmasac_policy.py
swarmbots/learn/algos/sac/tmasac_actor_heads.py
swarmbots/learn/algos/sac/tmasac_policy.py
swarmbots/learn/algos/world_modeling/__init__.py
swarmbots/learn/algos/world_modeling/base_wm_sampler.py
swarmbots/learn/algos/world_modeling/next_obs_pred_mixin.py
swarmbots/learn/algos/world_modeling/next_obs_pred_ppo_wrapper.py
swarmbots/learn/algos/world_modeling/ppo_wm_sampler.py
swarmbots/learn/algos/world_modeling/spr_mixin.py
swarmbots/learn/algos/world_modeling/spr_ppo_wrapper.py
swarmbots/learn/algos/world_modeling/transformer_transition_model.py
swarmbots/learn/algos/world_modeling/wm_recurrent_batch.py
swarmbots/learn/algos/world_modeling/wm_sampler_helper.py
swarmbots/learn/algos/xlstm/__init__.py
swarmbots/learn/algos/xlstm/head_utils.py
swarmbots/learn/algos/xlstm/temporal_utils.py
swarmbots/learn/algos/xlstm/mlstm/__init__.py
swarmbots/learn/algos/xlstm/mlstm/mlstm_cell.py
swarmbots/learn/algos/xlstm/mlstm/mlstm_temporal_sequence_model.py
swarmbots/learn/algos/xlstm/slstm/__init__.py
swarmbots/learn/algos/xlstm/slstm/slstm_cell.py
swarmbots/learn/algos/xlstm/slstm/slstm_temporal_sequence_model.py
swarmbots/learn/env_wrappers/__init__.py
swarmbots/learn/env_wrappers/feature_wise_obs_norm_wrapper.py
swarmbots/learn/env_wrappers/move_to_payload_global_obs_adapter.py
swarmbots/learn/env_wrappers/multi_scenario_vector_env.py
swarmbots/learn/env_wrappers/progress_guidance_ep_stats_wrapper.py
swarmbots/learn/env_wrappers/torch_feature_wise_obs_norm_tensor_ops.py
swarmbots/learn/env_wrappers/torch_feature_wise_obs_norm_wrapper.py
swarmbots/learn/env_wrappers/torch_normalize_reward_wrapper.py
swarmbots/learn/env_wrappers/torch_progress_guidance_ep_stats_wrapper.py
swarmbots/learn/env_wrappers/torch_record_episode_statistics_wrapper.py
swarmbots/learn/env_wrappers/torch_shuffle_agents_wrapper.py
swarmbots/learn/env_wrappers/torch_transition_obs_wrapper.py
swarmbots/learn/env_wrappers/worker_pool_async_vector_env.py
swarmbots/learn/env_wrappers/learn_wrappers/__init__.py
swarmbots/learn/env_wrappers/learn_wrappers/base_learn_env_wrapper.py
swarmbots/learn/env_wrappers/learn_wrappers/continuous_actions_learn_env_wrapper.py
swarmbots/learn/env_wrappers/learn_wrappers/swarm_bots_learn_env_wrapper.py
swarmbots/learn/env_wrappers/learn_wrappers/torch_env_wrapper.py
swarmbots/learn/nn_components/__init__.py
swarmbots/learn/nn_components/activations.py
swarmbots/learn/nn_components/custom_transformer_decoder_layer.py
swarmbots/learn/nn_components/deep_set.py
swarmbots/learn/nn_components/feed_forward.py
swarmbots/learn/nn_components/nn_init.py
swarmbots/learn/nn_components/pelu.py
swarmbots/learn/nn_components/popart.py
swarmbots/learn/nn_components/residual.py
swarmbots/learn/presets/__init__.py
swarmbots/learn/presets/policy_factory.py
swarmbots/learn/presets/tmasac.py
swarmbots/learn/presets/transformer.py
swarmbots/learn/presets/variants.py
swarmbots/learn/scheduling/__init__.py
swarmbots/learn/scheduling/auto_lr_updater.py
swarmbots/learn/scheduling/chainable_scheduler.py
swarmbots/learn/scheduling/cosine_scheduler.py
swarmbots/learn/scheduling/exponential_scheduler.py
swarmbots/learn/scheduling/linear_scheduler.py
swarmbots/learn/scheduling/scheduler_factory.py
swarmbots/learn/scheduling/schedulers.py
swarmbots/mjw_env/__init__.py
swarmbots/mjw_env/mjw_env_tensor_ops.py
swarmbots/mjw_env/mjw_kernels.py
swarmbots/mjw_env/mjw_live_recording.py
swarmbots/mjw_env/mjw_model_metadata.py
swarmbots/mjw_env/mjw_swarm_bots_vector_env.py
swarmbots/mjw_env/mjw_torch_quat.py
swarmbots/mjw_env/mjw_torch_utils.py
swarmbots/mjw_env/mujoco_utils.py
swarmbots/mjw_env/parameter_distributions.py
swarmbots/mjw_env/random_utils.py
swarmbots/mjw_env/scenarios/__init__.py
swarmbots/mjw_env/scenarios/base_mjw_scenario.py
swarmbots/mjw_env/scenarios/mjw_bridge_runtime.py
swarmbots/mjw_env/scenarios/mjw_bridge_scenario.py
swarmbots/mjw_env/scenarios/mjw_climb_runtime.py
swarmbots/mjw_env/scenarios/mjw_climb_scenario.py
swarmbots/mjw_env/scenarios/mjw_dual_payload_plane_runtime.py
swarmbots/mjw_env/scenarios/mjw_dual_payload_plane_scenario.py
swarmbots/mjw_env/scenarios/mjw_find_opening_runtime.py
swarmbots/mjw_env/scenarios/mjw_find_opening_scenario.py
swarmbots/mjw_env/scenarios/mjw_move_to_runtime.py
swarmbots/mjw_env/scenarios/mjw_move_to_scenario.py
swarmbots/mjw_env/scenarios/mjw_multi_payload_goal_runtime.py
swarmbots/mjw_env/scenarios/mjw_multi_payload_goal_scenario.py
swarmbots/mjw_env/scenarios/mjw_payload_plane_runtime.py
swarmbots/mjw_env/scenarios/mjw_payload_plane_scenario.py
swarmbots/mjw_env/scenarios/mjw_payload_step_runtime.py
swarmbots/mjw_env/scenarios/mjw_payload_step_scenario.py
swarmbots/mjw_env/scenarios/mjw_scenario_presets.py
swarmbots/mjw_env/scenarios/mjw_vertical_reach_runtime.py
swarmbots/mjw_env/scenarios/mjw_vertical_reach_scenario.py
swarmbots/mjw_env/scenarios/mjw_wall_runtime.py
swarmbots/mjw_env/scenarios/mjw_wall_scenario.py
swarmbots/mjw_env/swarm/__init__.py
swarmbots/mjw_env/swarm/mjw_homogeneous_swarm.py
swarmbots/mjw_env/swarm/mjw_swarm_config.py
swarmbots/mjw_env/swarm/unit.py
swarmbots/mjw_env/swarm/unit_config.py
swarmbots/scenario_presets/__init__.py
swarmbots/scenario_presets/move_to_goal_config.py
swarmbots/scenario_presets/multi_payload_goal.py
swarmbots/scenario_presets/scenario_obs_layouts.py
swarmbots/scenario_presets/scenario_presets_kwargs.py
swarmbots/utils/__init__.py
swarmbots/utils/connector_actions.py
swarmbots/utils/machine_specs.py
swarmbots/utils/mujoco_bootstrap.py
swarmbots/utils/mujoco_render_geoms.py
swarmbots/utils/mujoco_step_geoms.py
swarmbots/utils/recording_overlay.py
swarmbots/utils/recording_resolution.py
swarmbots/utils/torch_utils.py
tests/conftest.py
tests/test_action_metrics.py
tests/test_activation_factory_regression.py
tests/test_auto_lr_updater.py
tests/test_base_algorithm_eta.py
tests/test_base_algorithm_lifecycle.py
tests/test_base_sampler.py
tests/test_checkpointing_env_state.py
tests/test_discord_notifications.py
tests/test_entropy_utils.py
tests/test_environment_smoke.py
tests/test_evaluation.py
tests/test_evaluation_metadata.py
tests/test_frozen_evaluation.py
tests/test_gsde_action_dist.py
tests/test_gumbel_softmax_sign_magnitude_action_dist.py
tests/test_hybrid_action_dist_factory.py
tests/test_learning_training.py
tests/test_mat_dec_policy.py
tests/test_mat_encoder_feedforward_configs.py
tests/test_mat_encoder_layer.py
tests/test_mat_ind_policy.py
tests/test_mat_orig_policy.py
tests/test_mat_qcx_decoder.py
tests/test_mat_qcx_policy.py
tests/test_metrics_logger.py
tests/test_mjw_connection_tracking.py
tests/test_mjw_connector_kernels.py
tests/test_mjw_env_tensor_ops.py
tests/test_mjw_live_recording_renderer_reuse.py
tests/test_mjw_settled_resets.py
tests/test_mjw_wall_pass_reward.py
tests/test_multi_scenario_vector_env.py
tests/test_next_obs_pred_global.py
tests/test_obs_indices.py
tests/test_off_policy_replay.py
tests/test_policy_recording.py
tests/test_popart.py
tests/test_ppo_mappo_initialization.py
tests/test_ppo_masking.py
tests/test_ppo_parameter_lr_multipliers.py
tests/test_ppo_rollout_batch.py
tests/test_ppo_rollout_buffer_sizing.py
tests/test_ppo_sampler.py
tests/test_ppo_wm_sampler.py
tests/test_recording_recurrent.py
tests/test_recurrent_tbptt.py
tests/test_recurrent_tmasac.py
tests/test_recurrent_world_modeling.py
tests/test_registry.py
tests/test_release_metadata.py
tests/test_rollout_utils.py
tests/test_sac_action_sampling.py
tests/test_sac_algorithm.py
tests/test_sac_nop.py
tests/test_sac_tensor_ops.py
tests/test_same_step_pipeline.py
tests/test_scenario_obs_encoder.py
tests/test_schedulers.py
tests/test_summary_statistics.py
tests/test_tensor_conversion.py
tests/test_tmasac_policy.py
tests/test_torch_device.py
tests/test_torch_feature_wise_obs_norm_tensor_ops.py
tests/test_torch_feature_wise_obs_norm_wrapper.py
tests/test_torch_running_mean_std.py
tests/test_torch_shuffle_agents_wrapper.py
tests/test_transformer_initialization.py
tests/test_xlstm_cells.py