Path	Lines of Code
local_test.sh	20
tests/dummy.py	4
tests/test_examples.py	88
tests/test_jax_api.py	154
tests/test_numpy_api.py	150
tests/test_tensorflow_api.py	182
tests/test_torch_api.py	165