numpy
opt-einsum
tensorflow-gpu>=2.0.0
pyamg
