diff --git a/petals/__init__.py b/src/petals/__init__.py
similarity index 100%
rename from petals/__init__.py
rename to src/petals/__init__.py
diff --git a/petals/bloom/__init__.py b/src/petals/bloom/__init__.py
similarity index 100%
rename from petals/bloom/__init__.py
rename to src/petals/bloom/__init__.py
diff --git a/petals/bloom/block.py b/src/petals/bloom/block.py
similarity index 100%
rename from petals/bloom/block.py
rename to src/petals/bloom/block.py
diff --git a/petals/bloom/from_pretrained.py b/src/petals/bloom/from_pretrained.py
similarity index 100%
rename from petals/bloom/from_pretrained.py
rename to src/petals/bloom/from_pretrained.py
diff --git a/petals/bloom/model.py b/src/petals/bloom/model.py
similarity index 100%
rename from petals/bloom/model.py
rename to src/petals/bloom/model.py
diff --git a/petals/bloom/ops.py b/src/petals/bloom/ops.py
similarity index 100%
rename from petals/bloom/ops.py
rename to src/petals/bloom/ops.py
diff --git a/petals/client/__init__.py b/src/petals/client/__init__.py
similarity index 100%
rename from petals/client/__init__.py
rename to src/petals/client/__init__.py
diff --git a/petals/client/inference_session.py b/src/petals/client/inference_session.py
similarity index 100%
rename from petals/client/inference_session.py
rename to src/petals/client/inference_session.py
diff --git a/petals/client/remote_forward_backward.py b/src/petals/client/remote_forward_backward.py
similarity index 100%
rename from petals/client/remote_forward_backward.py
rename to src/petals/client/remote_forward_backward.py
diff --git a/petals/client/remote_generation.py b/src/petals/client/remote_generation.py
similarity index 100%
rename from petals/client/remote_generation.py
rename to src/petals/client/remote_generation.py
diff --git a/petals/client/remote_model.py b/src/petals/client/remote_model.py
similarity index 100%
rename from petals/client/remote_model.py
rename to src/petals/client/remote_model.py
diff --git a/petals/client/remote_sequential.py b/src/petals/client/remote_sequential.py
similarity index 100%
rename from petals/client/remote_sequential.py
rename to src/petals/client/remote_sequential.py
diff --git a/petals/client/sequence_manager.py b/src/petals/client/sequence_manager.py
similarity index 100%
rename from petals/client/sequence_manager.py
rename to src/petals/client/sequence_manager.py
diff --git a/petals/client/sequential_autograd.py b/src/petals/client/sequential_autograd.py
similarity index 100%
rename from petals/client/sequential_autograd.py
rename to src/petals/client/sequential_autograd.py
diff --git a/petals/client/spending_policy.py b/src/petals/client/spending_policy.py
similarity index 100%
rename from petals/client/spending_policy.py
rename to src/petals/client/spending_policy.py
diff --git a/petals/constants.py b/src/petals/constants.py
similarity index 100%
rename from petals/constants.py
rename to src/petals/constants.py
diff --git a/petals/data_structures.py b/src/petals/data_structures.py
similarity index 100%
rename from petals/data_structures.py
rename to src/petals/data_structures.py
diff --git a/petals/dht_utils.py b/src/petals/dht_utils.py
similarity index 100%
rename from petals/dht_utils.py
rename to src/petals/dht_utils.py
diff --git a/petals/server/__init__.py b/src/petals/server/__init__.py
similarity index 100%
rename from petals/server/__init__.py
rename to src/petals/server/__init__.py
diff --git a/petals/server/backend.py b/src/petals/server/backend.py
similarity index 100%
rename from petals/server/backend.py
rename to src/petals/server/backend.py
diff --git a/petals/server/block_selection.py b/src/petals/server/block_selection.py
similarity index 100%
rename from petals/server/block_selection.py
rename to src/petals/server/block_selection.py
diff --git a/petals/server/cache.py b/src/petals/server/cache.py
similarity index 100%
rename from petals/server/cache.py
rename to src/petals/server/cache.py
diff --git a/petals/server/handler.py b/src/petals/server/handler.py
similarity index 100%
rename from petals/server/handler.py
rename to src/petals/server/handler.py
diff --git a/petals/server/runtime.py b/src/petals/server/runtime.py
similarity index 100%
rename from petals/server/runtime.py
rename to src/petals/server/runtime.py
diff --git a/petals/server/server.py b/src/petals/server/server.py
similarity index 100%
rename from petals/server/server.py
rename to src/petals/server/server.py
diff --git a/petals/server/task_pool.py b/src/petals/server/task_pool.py
similarity index 100%
rename from petals/server/task_pool.py
rename to src/petals/server/task_pool.py
diff --git a/petals/server/task_prioritizer.py b/src/petals/server/task_prioritizer.py
similarity index 100%
rename from petals/server/task_prioritizer.py
rename to src/petals/server/task_prioritizer.py
diff --git a/petals/server/throughput.py b/src/petals/server/throughput.py
similarity index 100%
rename from petals/server/throughput.py
rename to src/petals/server/throughput.py
diff --git a/petals/src/__init__.py b/src/petals/src/__init__.py
similarity index 100%
rename from petals/src/__init__.py
rename to src/petals/src/__init__.py
diff --git a/petals/src/bloom/__init__.py b/src/petals/src/bloom/__init__.py
similarity index 100%
rename from petals/src/bloom/__init__.py
rename to src/petals/src/bloom/__init__.py
diff --git a/petals/src/bloom/block.py b/src/petals/src/bloom/block.py
similarity index 100%
rename from petals/src/bloom/block.py
rename to src/petals/src/bloom/block.py
diff --git a/petals/src/bloom/from_pretrained.py b/src/petals/src/bloom/from_pretrained.py
similarity index 100%
rename from petals/src/bloom/from_pretrained.py
rename to src/petals/src/bloom/from_pretrained.py
diff --git a/petals/src/bloom/model.py b/src/petals/src/bloom/model.py
similarity index 100%
rename from petals/src/bloom/model.py
rename to src/petals/src/bloom/model.py
diff --git a/petals/src/bloom/ops.py b/src/petals/src/bloom/ops.py
similarity index 100%
rename from petals/src/bloom/ops.py
rename to src/petals/src/bloom/ops.py
diff --git a/petals/src/client/__init__.py b/src/petals/src/client/__init__.py
similarity index 100%
rename from petals/src/client/__init__.py
rename to src/petals/src/client/__init__.py
diff --git a/petals/src/client/inference_session.py b/src/petals/src/client/inference_session.py
similarity index 100%
rename from petals/src/client/inference_session.py
rename to src/petals/src/client/inference_session.py
diff --git a/petals/src/client/remote_forward_backward.py b/src/petals/src/client/remote_forward_backward.py
similarity index 100%
rename from petals/src/client/remote_forward_backward.py
rename to src/petals/src/client/remote_forward_backward.py
diff --git a/petals/src/client/remote_generation.py b/src/petals/src/client/remote_generation.py
similarity index 100%
rename from petals/src/client/remote_generation.py
rename to src/petals/src/client/remote_generation.py
diff --git a/petals/src/client/remote_model.py b/src/petals/src/client/remote_model.py
similarity index 100%
rename from petals/src/client/remote_model.py
rename to src/petals/src/client/remote_model.py
diff --git a/petals/src/client/remote_sequential.py b/src/petals/src/client/remote_sequential.py
similarity index 100%
rename from petals/src/client/remote_sequential.py
rename to src/petals/src/client/remote_sequential.py
diff --git a/petals/src/client/sequence_manager.py b/src/petals/src/client/sequence_manager.py
similarity index 100%
rename from petals/src/client/sequence_manager.py
rename to src/petals/src/client/sequence_manager.py
diff --git a/petals/src/client/sequential_autograd.py b/src/petals/src/client/sequential_autograd.py
similarity index 100%
rename from petals/src/client/sequential_autograd.py
rename to src/petals/src/client/sequential_autograd.py
diff --git a/petals/src/client/spending_policy.py b/src/petals/src/client/spending_policy.py
similarity index 100%
rename from petals/src/client/spending_policy.py
rename to src/petals/src/client/spending_policy.py
diff --git a/petals/src/constants.py b/src/petals/src/constants.py
similarity index 100%
rename from petals/src/constants.py
rename to src/petals/src/constants.py
diff --git a/petals/src/data_structures.py b/src/petals/src/data_structures.py
similarity index 100%
rename from petals/src/data_structures.py
rename to src/petals/src/data_structures.py
diff --git a/petals/src/dht_utils.py b/src/petals/src/dht_utils.py
similarity index 100%
rename from petals/src/dht_utils.py
rename to src/petals/src/dht_utils.py
diff --git a/petals/src/server/__init__.py b/src/petals/src/server/__init__.py
similarity index 100%
rename from petals/src/server/__init__.py
rename to src/petals/src/server/__init__.py
diff --git a/petals/src/server/backend.py b/src/petals/src/server/backend.py
similarity index 100%
rename from petals/src/server/backend.py
rename to src/petals/src/server/backend.py
diff --git a/petals/src/server/block_selection.py b/src/petals/src/server/block_selection.py
similarity index 100%
rename from petals/src/server/block_selection.py
rename to src/petals/src/server/block_selection.py
diff --git a/petals/src/server/cache.py b/src/petals/src/server/cache.py
similarity index 100%
rename from petals/src/server/cache.py
rename to src/petals/src/server/cache.py
diff --git a/petals/src/server/handler.py b/src/petals/src/server/handler.py
similarity index 100%
rename from petals/src/server/handler.py
rename to src/petals/src/server/handler.py
diff --git a/petals/src/server/runtime.py b/src/petals/src/server/runtime.py
similarity index 100%
rename from petals/src/server/runtime.py
rename to src/petals/src/server/runtime.py
diff --git a/petals/src/server/server.py b/src/petals/src/server/server.py
similarity index 100%
rename from petals/src/server/server.py
rename to src/petals/src/server/server.py
diff --git a/petals/src/server/task_pool.py b/src/petals/src/server/task_pool.py
similarity index 100%
rename from petals/src/server/task_pool.py
rename to src/petals/src/server/task_pool.py
diff --git a/petals/src/server/task_prioritizer.py b/src/petals/src/server/task_prioritizer.py
similarity index 100%
rename from petals/src/server/task_prioritizer.py
rename to src/petals/src/server/task_prioritizer.py
diff --git a/petals/src/server/throughput.py b/src/petals/src/server/throughput.py
similarity index 100%
rename from petals/src/server/throughput.py
rename to src/petals/src/server/throughput.py
diff --git a/petals/src/utils/__init__.py b/src/petals/src/utils/__init__.py
similarity index 100%
rename from petals/src/utils/__init__.py
rename to src/petals/src/utils/__init__.py
diff --git a/petals/src/utils/convert_8bit.py b/src/petals/src/utils/convert_8bit.py
similarity index 100%
rename from petals/src/utils/convert_8bit.py
rename to src/petals/src/utils/convert_8bit.py
diff --git a/petals/src/utils/generation_algorithms.py b/src/petals/src/utils/generation_algorithms.py
similarity index 100%
rename from petals/src/utils/generation_algorithms.py
rename to src/petals/src/utils/generation_algorithms.py
diff --git a/petals/src/utils/generation_constraints.py b/src/petals/src/utils/generation_constraints.py
similarity index 100%
rename from petals/src/utils/generation_constraints.py
rename to src/petals/src/utils/generation_constraints.py
diff --git a/petals/src/utils/misc.py b/src/petals/src/utils/misc.py
similarity index 100%
rename from petals/src/utils/misc.py
rename to src/petals/src/utils/misc.py
diff --git a/petals/utils/__init__.py b/src/petals/utils/__init__.py
similarity index 100%
rename from petals/utils/__init__.py
rename to src/petals/utils/__init__.py
diff --git a/petals/utils/convert_8bit.py b/src/petals/utils/convert_8bit.py
similarity index 100%
rename from petals/utils/convert_8bit.py
rename to src/petals/utils/convert_8bit.py
diff --git a/petals/utils/generation_algorithms.py b/src/petals/utils/generation_algorithms.py
similarity index 100%
rename from petals/utils/generation_algorithms.py
rename to src/petals/utils/generation_algorithms.py
diff --git a/petals/utils/generation_constraints.py b/src/petals/utils/generation_constraints.py
similarity index 100%
rename from petals/utils/generation_constraints.py
rename to src/petals/utils/generation_constraints.py
diff --git a/petals/utils/misc.py b/src/petals/utils/misc.py
similarity index 100%
rename from petals/utils/misc.py
rename to src/petals/utils/misc.py