absl-py~=1.4.0
apache-beam>=2.47.0
numpy~=1.21
tensorflow~=2.12.0
tensorflow-datasets~=4.9.0

[dev]
pytest
pytest-xdist
