PyPI page
Home page
Author:
Lelouch
License:
Summary:
TensorWrap: A high level interface aimed towards rapid prototyping and intuitive workflow for JAX.
Latest version:
0.0.1.8
Required dependencies:
dill
|
jax
|
jaxlib
|
jaxtyping
|
matplotlib
|
numpy
|
optax
|
pandas
|
tensorflow
|
tensorflow-datasets
|
termcolor
Optional dependencies:
atari-py
|
clu
|
gym
|
jaxlib
|
jraph
|
ml-collections
|
mypy
|
opencv-python
|
pytest
|
pytest-cov
|
pytest-custom_exit_code
|
pytest-xdist
|
pytype
|
sentencepiece
|
tensorflow
|
tensorflow_datasets
|
tensorflow_text
|
torch
Downloads last day:
8
Downloads last week:
31
Downloads last month:
87