3b115c1073
* Add NN for pytorch backend * Add PyTorch nn functions * Add PyTorch nn functions * Adds PyTorch Backend * Adds PyTorch Backend * Adds PyTorch nn backend * Add PyTorch backend funcitons * Add Torch NN Backend * Add Torch NN Backend * Add Torch NN Backend
14 lines
181 B
Plaintext
14 lines
181 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 |