[FIX] Python implementation of TensorCoFi support for multiple contexts (Not the Java one)
[FIX] Java implementation of TensorCoFi support for multiple contexts. Need testing