Update neural_network_future.py

Temporarily manually set jit_compile to True to check compatibility.

Author: david-thrower

cerebros/neuralnetworkfuture/neural_network_future.py

@@ -341,7 +341,7 @@ def compile_neural_network(self):
                         weight_decay=0.004,  # Add weight decay parameter
                         gradient_accumulation_steps=self.gradient_accumulation_steps
                     ),
-                    jit_compile=jit_compile)
+                    jit_compile=Tree) # jit_compile)
         elif self.gradient_accumulation_steps == 1:
             self.materialized_neural_network.compile(
                     loss=self.loss,