Rename the kwargs parameter in jitted functions

kernel/modelverse_jit/jit.py

@@ -1,7 +1,7 @@
 import modelverse_kernel.primitives as primitive_functions
 import modelverse_jit.tree_ir as tree_ir
 
-KWARGS_PARAMETER_NAME = "remainder"
+KWARGS_PARAMETER_NAME = "kwargs"
 """The name of the kwargs parameter in jitted functions."""
 
 class JitCompilationFailedException(Exception):