b14eaaf5da
* Add PyTorch numpy functionality * Add dtype conversion * Partial fix for PyTorch numpy tests * small logic fix * Revert numpy_test * Add tensor conversion to numpy * Fix some arithmetic tests * Fix some torch functions for numpy compatibility * Fix pytorch ops for numpy compatibility, add TODOs * Fix formatting * Implement nits and fix dtype standardization * Add pytest skipif decorator and fix nits * Fix formatting and rename dtypes map * Split tests by backend * Merge space * Fix dtype issues from new type checking * Implement torch.full and torch.full_like numpy compatible * Implements logspace and linspace with tensor support for start and stop * Replace len of shape with ndim * Fix formatting * Implement torch.trace * Implement eye k diagonal arg * Implement torch.tri * Fix formatting issues * Fix torch.take dimensionality * Add split functionality * Revert torch.eye implementation to prevent conflict * Implement all padding modes * Adds torch image resizing and torchvision dependency. * Fix conditional syntax * Make torchvision import optional
15 lines
201 B
Plaintext
15 lines
201 B
Plaintext
tensorflow
|
|
# TODO: Use Torch CPU
|
|
# Remove after resolving Cuda version differences with TF
|
|
torch>=2.0.1+cpu
|
|
torchvision>=0.15.1
|
|
jax[cpu]
|
|
namex
|
|
black>=22
|
|
flake8
|
|
isort
|
|
pytest
|
|
pandas
|
|
absl-py
|
|
requests
|
|
h5py |