699e4c3174
* Begin adding testing for tensorboard callback * Begin protobuf areas * Sharing WIP PR * Rework init * Add start/stop profiling to backend * Add image/histogram summaries * Model tracing * Full tf working * Enable jax graphs * Fix cifar100 dataset * Finish jax integration checkpoint without tracing * Formatting * All tests passing locally * Model path protected, switch to _model * Remove non-deterministic variable-n tag test * Formatting * Remove profiling jax * Remove tensorboard ops * Remove profiling * File utils fix? * Revert "File utils fix?" This reverts commit a48ba39a9c51c1e39f4509f795fdc2c316a1da3b.
21 lines
256 B
Plaintext
21 lines
256 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
|
|
protobuf
|
|
google
|
|
tensorboard-plugin-profile
|
|
rich
|
|
build
|