├── .github └── workflows │ └── python-app.yml ├── .gitignore ├── LICENSE ├── README.md ├── agentlace ├── __init__.py ├── action.py ├── data │ ├── __init__.py │ ├── data_store.py │ ├── jaxrl_data_store.py │ ├── rlds_writer.py │ ├── sampler.py │ ├── tf_agents_episode_buffer.py │ ├── tfds.py │ └── trajectory_buffer.py ├── gym_env.py ├── inference.py ├── internal │ ├── __init__.py │ └── utils.py ├── trainer.py └── zmq_wrapper │ ├── __init__.py │ ├── broadcast.py │ ├── pair.py │ ├── pipeline.py │ └── req_rep.py ├── docs └── agentlace_logo.png ├── examples ├── action_streamer.py ├── async_learner_actor.py ├── gym_cartpole_env.py ├── jaxrl_m_common.py ├── omnigibson_env.py ├── robosuite_env.py ├── run_data_store.py └── train_online.py ├── setup.py └── tests ├── test_action.py ├── test_all.py ├── test_data_store.py ├── test_image.png ├── test_inference.py ├── test_tfds.py └── test_trainer.py /.github/workflows/python-app.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/.github/workflows/python-app.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/README.md -------------------------------------------------------------------------------- /agentlace/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/__init__.py -------------------------------------------------------------------------------- /agentlace/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/action.py -------------------------------------------------------------------------------- /agentlace/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /agentlace/data/data_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/data/data_store.py -------------------------------------------------------------------------------- /agentlace/data/jaxrl_data_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/data/jaxrl_data_store.py -------------------------------------------------------------------------------- /agentlace/data/rlds_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/data/rlds_writer.py -------------------------------------------------------------------------------- /agentlace/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/data/sampler.py -------------------------------------------------------------------------------- /agentlace/data/tf_agents_episode_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/data/tf_agents_episode_buffer.py -------------------------------------------------------------------------------- /agentlace/data/tfds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/data/tfds.py -------------------------------------------------------------------------------- /agentlace/data/trajectory_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/data/trajectory_buffer.py -------------------------------------------------------------------------------- /agentlace/gym_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/gym_env.py -------------------------------------------------------------------------------- /agentlace/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/inference.py -------------------------------------------------------------------------------- /agentlace/internal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /agentlace/internal/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/internal/utils.py -------------------------------------------------------------------------------- /agentlace/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/trainer.py -------------------------------------------------------------------------------- /agentlace/zmq_wrapper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /agentlace/zmq_wrapper/broadcast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/zmq_wrapper/broadcast.py -------------------------------------------------------------------------------- /agentlace/zmq_wrapper/pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/zmq_wrapper/pair.py -------------------------------------------------------------------------------- /agentlace/zmq_wrapper/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/zmq_wrapper/pipeline.py -------------------------------------------------------------------------------- /agentlace/zmq_wrapper/req_rep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/agentlace/zmq_wrapper/req_rep.py -------------------------------------------------------------------------------- /docs/agentlace_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/docs/agentlace_logo.png -------------------------------------------------------------------------------- /examples/action_streamer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/action_streamer.py -------------------------------------------------------------------------------- /examples/async_learner_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/async_learner_actor.py -------------------------------------------------------------------------------- /examples/gym_cartpole_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/gym_cartpole_env.py -------------------------------------------------------------------------------- /examples/jaxrl_m_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/jaxrl_m_common.py -------------------------------------------------------------------------------- /examples/omnigibson_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/omnigibson_env.py -------------------------------------------------------------------------------- /examples/robosuite_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/robosuite_env.py -------------------------------------------------------------------------------- /examples/run_data_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/run_data_store.py -------------------------------------------------------------------------------- /examples/train_online.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/examples/train_online.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/tests/test_action.py -------------------------------------------------------------------------------- /tests/test_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/tests/test_all.py -------------------------------------------------------------------------------- /tests/test_data_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/tests/test_data_store.py -------------------------------------------------------------------------------- /tests/test_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/tests/test_image.png -------------------------------------------------------------------------------- /tests/test_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/tests/test_inference.py -------------------------------------------------------------------------------- /tests/test_tfds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/tests/test_tfds.py -------------------------------------------------------------------------------- /tests/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/youliangtan/agentlace/HEAD/tests/test_trainer.py --------------------------------------------------------------------------------