diff --git a/src/elisa/__init__.py b/src/elisa/__init__.py
index fae8b82..c375363 100644
--- a/src/elisa/__init__.py
+++ b/src/elisa/__init__.py
@@ -33,5 +33,4 @@
 )
 
 jax_enable_x64(True)
-# set_jax_platform('cpu')
 set_cpu_cores(4)