Running into some annoying bugs tackling #136, so I wanted to see if updating libraries might help. Also removed Python 3.6 (due to end of life) and added Python 3.9 compatibility (since the TFQ nightly build supports TF 2.7.0, and that supports 3.9). Part of #144, that will be closed when TFQ 0.6.0 is out of nightly.
TFQ 0.5.1 -> 0.6.0.dev20220125, TF 2.4.1 -> 2.7.0, TFP 0.12.0 -> 0.15.0
Running into some annoying bugs tackling #136, so I wanted to see if updating libraries might help. Also removed Python 3.6 (due to end of life) and added Python 3.9 compatibility (since the TFQ nightly build supports TF 2.7.0, and that supports 3.9). Part of #144, that will be closed when TFQ 0.6.0 is out of nightly.