bf734b0eec
* Add torch backend - random * Add Random Unit Test * Adds Random Unit Test * Random Test * Adds Random Test
15 lines
187 B
Plaintext
15 lines
187 B
Plaintext
tensorflow
|
|
# TODO: Use Torch CPU
|
|
# Remove after resolving Cuda version differences with TF
|
|
torch>=2.0.1+cpu
|
|
jax[cpu]
|
|
namex
|
|
black>=22
|
|
flake8
|
|
isort
|
|
pytest
|
|
pandas
|
|
absl-py
|
|
requests
|
|
h5py
|
|
torch |