torch>=2.0

[all]
esmoe[torch]
esmoe[tensorflow]
esmoe[jax]
esmoe[flax]
esmoe[keras]
esmoe[yolo]

[flax]
flax>=0.10
jax>=0.4.35

[jax]
jax>=0.4.35

[keras]
keras>=3.5

[keras3]
esmoe[keras]

[nnx]
esmoe[flax]

[pytorch]
esmoe[torch]

[tensorflow]
tensorflow>=2.17

[tf]
esmoe[tensorflow]

[torch]
torch>=2.0

[ultralytics]
esmoe[yolo]

[yolo]
ultralytics>=8.3
