LICENSE
README.md
pyproject.toml
tests/test_action_mask.py
tests/test_algos_clock.py
tests/test_algos_contract.py
tests/test_algos_extension.py
tests/test_algos_no_ttnn.py
tests/test_algos_registry.py
tests/test_algos_step_modes.py
tests/test_algos_transition.py
tests/test_import_isolation.py
tests/test_obs_coercion.py
tests/test_obs_leaves.py
tests/test_obs_split.py
tests/test_public_api.py
tests/test_registries_are_live.py
tests/test_ttnn_version.py
tt_rl.egg-info/PKG-INFO
tt_rl.egg-info/SOURCES.txt
tt_rl.egg-info/dependency_links.txt
tt_rl.egg-info/entry_points.txt
tt_rl.egg-info/requires.txt
tt_rl.egg-info/top_level.txt
ttrl/__init__.py
ttrl/algos/__init__.py
ttrl/algos/base.py
ttrl/algos/checkpoint.py
ttrl/algos/checks.py
ttrl/algos/clock.py
ttrl/algos/config.py
ttrl/algos/gaussian.py
ttrl/algos/grads.py
ttrl/algos/mask.py
ttrl/algos/offpolicy.py
ttrl/algos/registry.py
ttrl/algos/scaling.py
ttrl/algos/dqn/__init__.py
ttrl/algos/dqn/agent.py
ttrl/algos/dqn/config.py
ttrl/algos/dqn/losses.py
ttrl/algos/ppo/__init__.py
ttrl/algos/ppo/agent.py
ttrl/algos/ppo/boundaries.py
ttrl/algos/ppo/build.py
ttrl/algos/ppo/config.py
ttrl/algos/ppo/learn.py
ttrl/algos/ppo/loss.py
ttrl/algos/ppo/partial.py
ttrl/algos/ppo/policy.py
ttrl/algos/ppo/recurrent.py
ttrl/algos/ppo/state.py
ttrl/algos/sac/__init__.py
ttrl/algos/sac/agent.py
ttrl/algos/sac/config.py
ttrl/algos/sac/losses.py
ttrl/algos/sac/policy.py
ttrl/algos/sac/step.py
ttrl/algos/sac/targets.py
ttrl/buffers/__init__.py
ttrl/buffers/boundaries.py
ttrl/buffers/checks.py
ttrl/buffers/indexing.py
ttrl/buffers/openepisodes.py
ttrl/buffers/persist.py
ttrl/buffers/replay.py
ttrl/buffers/replaybatch.py
ttrl/buffers/rollout.py
ttrl/buffers/samplers.py
ttrl/buffers/sampling.py
ttrl/buffers/seqbatch.py
ttrl/buffers/sequences.py
ttrl/buffers/types.py
ttrl/buffers/window.py
ttrl/buffers/writes.py
ttrl/cli/__init__.py
ttrl/cli/__main__.py
ttrl/cli/doctor.py
ttrl/cli/logs.py
ttrl/cli/report.py
ttrl/cli/train.py
ttrl/cli/probes/__init__.py
ttrl/cli/probes/_ttnn.py
ttrl/cli/probes/chip_report.py
ttrl/cli/probes/chips.py
ttrl/cli/probes/extras.py
ttrl/cli/probes/hardware.py
ttrl/cli/probes/traps.py
ttrl/cli/probes/versions.py
ttrl/deploy/__init__.py
ttrl/deploy/artifact.py
ttrl/deploy/conv.py
ttrl/deploy/export.py
ttrl/deploy/graph.py
ttrl/deploy/policies.py
ttrl/deploy/read.py
ttrl/deploy/towers.py
ttrl/deploy/transform.py
ttrl/engine/__init__.py
ttrl/engine/autograd.py
ttrl/engine/carry.py
ttrl/engine/config.py
ttrl/engine/layers.py
ttrl/engine/recurrent.py
ttrl/engine/trace.py
ttrl/engine/conv/__init__.py
ttrl/engine/conv/backward.py
ttrl/engine/conv/config.py
ttrl/engine/conv/forward.py
ttrl/engine/conv/geometry.py
ttrl/engine/conv/module.py
ttrl/engine/conv/packing.py
ttrl/engine/conv/transform.py
ttrl/engine/conv/weights.py
ttrl/engine/device/__init__.py
ttrl/engine/device/acquire.py
ttrl/engine/device/liveness.py
ttrl/engine/device/profile.py
ttrl/engine/device/scope.py
ttrl/engine/mesh/__init__.py
ttrl/engine/mesh/devices.py
ttrl/engine/mesh/open.py
ttrl/engine/mesh/resume.py
ttrl/engine/mesh/sync.py
ttrl/engine/mesh/topology.py
ttrl/engine/nn/__init__.py
ttrl/engine/nn/containers.py
ttrl/engine/nn/discovery.py
ttrl/engine/nn/module.py
ttrl/engine/ops/__init__.py
ttrl/engine/ops/broadcast.py
ttrl/engine/ops/elementwise.py
ttrl/engine/ops/functional.py
ttrl/engine/ops/matmul.py
ttrl/engine/ops/reduce.py
ttrl/engine/ops/select.py
ttrl/engine/ops/shape.py
ttrl/engine/ops/unary.py
ttrl/engine/optim/__init__.py
ttrl/engine/optim/adam.py
ttrl/engine/optim/base.py
ttrl/engine/optim/checkpoint.py
ttrl/engine/optim/scalars.py
ttrl/engine/step/__init__.py
ttrl/engine/step/core.py
ttrl/engine/step/inputs.py
ttrl/engine/step/metrics.py
ttrl/engine/tensor/__init__.py
ttrl/engine/tensor/arena.py
ttrl/engine/tensor/precision.py
ttrl/engine/tensor/transfer.py
ttrl/envs/__init__.py
ttrl/envs/_children.py
ttrl/envs/_worker.py
ttrl/envs/base.py
ttrl/envs/bounds.py
ttrl/envs/check.py
ttrl/envs/check_clauses.py
ttrl/envs/episodes.py
ttrl/envs/gym_adapter.py
ttrl/envs/mujoco.py
ttrl/envs/obs_batch.py
ttrl/envs/registry.py
ttrl/envs/render.py
ttrl/envs/running.py
ttrl/envs/vec_normalize.py
ttrl/envs/vec_wrapper.py
ttrl/envs/vector.py
ttrl/envs/vector_async.py
ttrl/envs/atari/__init__.py
ttrl/envs/atari/emulator.py
ttrl/envs/atari/wrappers.py
ttrl/envs/wrappers/__init__.py
ttrl/envs/wrappers/control.py
ttrl/envs/wrappers/core.py
ttrl/envs/wrappers/leaf.py
ttrl/envs/wrappers/observation.py
ttrl/envs/wrappers/privileged.py
ttrl/envs/wrappers/reward.py
ttrl/nets/__init__.py
ttrl/nets/acting.py
ttrl/nets/build.py
ttrl/nets/choice.py
ttrl/nets/encoders.py
ttrl/nets/fusion.py
ttrl/nets/heads.py
ttrl/nets/models.py
ttrl/nets/multi.py
ttrl/nets/obs.py
ttrl/nets/recurrent.py
ttrl/nets/registry.py
ttrl/nets/towers.py
ttrl/spaces/__init__.py
ttrl/spaces/action.py
ttrl/spaces/fields.py
ttrl/spaces/group.py
ttrl/spaces/mask.py
ttrl/spaces/obs.py
ttrl/spaces/observation.py
ttrl/spaces/refusals.py
ttrl/train/__init__.py
ttrl/train/builders.py
ttrl/train/cadence.py
ttrl/train/device.py
ttrl/train/eval_checkpoint.py
ttrl/train/eval_metrics.py
ttrl/train/evaluate.py
ttrl/train/film.py
ttrl/train/gitstate.py
ttrl/train/hooks.py
ttrl/train/logging.py
ttrl/train/loop.py
ttrl/train/periodic.py
ttrl/train/provenance.py
ttrl/train/record.py
ttrl/train/record_env.py
ttrl/train/replicas.py
ttrl/train/restore.py
ttrl/train/resume.py
ttrl/train/resume_env.py
ttrl/train/run.py
ttrl/train/scoping.py
ttrl/train/snapshot.py
ttrl/train/stop.py
ttrl/train/teardown.py
ttrl/train/throughput.py
ttrl/train/trackers.py
ttrl/train/trainer.py
ttrl/train/video.py
ttrl/train/checkpoint/__init__.py
ttrl/train/checkpoint/atomic.py
ttrl/train/checkpoint/best.py
ttrl/train/checkpoint/file.py
ttrl/train/checkpoint/history.py
ttrl/train/checkpoint/read.py
ttrl/train/checkpoint/retype.py
ttrl/train/checkpoint/specs.py
ttrl/train/checkpoint/tree.py
ttrl/train/checkpoint/types.py
ttrl/train/config/__init__.py
ttrl/train/config/coerce.py
ttrl/train/config/device.py
ttrl/train/config/env.py
ttrl/train/config/horizon.py
ttrl/train/config/loader.py
ttrl/train/config/overrides.py
ttrl/train/config/schema.py