diff --git a/src/main.py b/src/main.py
index ef7c36e..ac40465 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,4 +1,9 @@
+import logging
+
 import jax
 
+logger = logging.getLogger(__name__)
+
 if __name__ == "__main__":
-    print(jax.devices())
+    logging.basicConfig(level=logging.INFO)
+    logger.info(f"JAX devices: {jax.devices()}")