keras/requirements.txt

# Tensorflow.
tensorflow

# Torch.
# TODO: Use Torch CPU, remove after resolving Cuda version differences with TF
torch>=2.0.1+cpu
torchvision>=0.15.1

# Jax.
jax[cpu]

# Common deps.
-r requirements-common.txt
Merge branch 'main' of github.com:keras-team/keras-core 2023-07-16 23:15:32 +00:00			`# Tensorflow.`
Initial draft of the codebase. 2023-04-09 19:21:45 +00:00			`tensorflow`
Merge branch 'main' of github.com:keras-team/keras-core 2023-07-16 23:15:32 +00:00
			`# Torch.`
			`# TODO: Use Torch CPU, remove after resolving Cuda version differences with TF`
Adds Random Unit Tests (#174) * Add torch backend - random * Add Random Unit Test * Adds Random Unit Test * Random Test * Adds Random Test 2023-05-16 05:21:36 +00:00			`torch>=2.0.1+cpu`
Adds torch image resizing and torchvision dependency. (#220) * 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 2023-05-30 16:19:01 +00:00			`torchvision>=0.15.1`
Merge branch 'main' of github.com:keras-team/keras-core 2023-07-16 23:15:32 +00:00
			`# Jax.`
Add some numpy ops (#1) * Add numpy ops (initial batch) and some config * Add unit test * fix call * Revert "fix call" This reverts commit 6748ad183029ff4b97317b77ceed8661916bb9a0. * full unit test coverage * fix setup.py 2023-04-12 18:31:58 +00:00			`jax[cpu]`
Merge branch 'main' of github.com:keras-team/keras-core 2023-07-16 23:15:32 +00:00
			`# Common deps.`
			`-r requirements-common.txt`