Update tensorflow to >=2.16

Closes #84. I tested the changes by running the `nn-classification` workflow on a small dataset and compared the results to the previous version - all seemed fine.