pyspark<4.0.0,>=3.4.0
pandas<3.0.0,>=1.3.4
networkx<3.0.0,>=2.6.3
pyfarmhash<0.4.0,>=0.3.2
keras<4.0.0,>=3.0.0
keras-tuner<2.0.0,>=1.4.0
numpy<2.0.0,>=1.22.0
tensorflow<3.0.0,>=2.16.0
dill<1.0.0,>=0.3.0

[jax]
jax>=0.4.0
jaxlib>=0.4.0

[torch]
torch>=2.0.0
