Not all calls to tf could be removed yet (specifically tf.random and tf.shape). This requires more modifications, so I have a follow-up PR for this, that allows switching between numpy and tensorflow.
Refactoring was done incrementally to check for instance for pytest performance (afaics, performance remains the same).
Closes #39
experimental_enable_numpy_behavior-- could be added if needed.tfcould be removed yet (specificallytf.randomandtf.shape). This requires more modifications, so I have a follow-up PR for this, that allows switching between numpy and tensorflow.