mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 336957861
This commit is contained in:
Jake VanderPlas 2020-10-13 14:36:16 -07:00 committed by jax authors
parent 83d011515d
commit 484ec3e2d6
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
jax/BUILD
View File

@ -95,6 +95,13 @@ pytype_library(
deps = [":jax"],
)
pytype_library(
name = "general_map",
srcs = ["experimental/general_map.py"],
srcs_version = "PY3",
deps = [":jax"],
)
pytype_library(
name = "jet",
srcs = ["experimental/jet.py"],
@ -102,6 +109,13 @@ pytype_library(
deps = [":jax"],
)
pytype_library(
name = "experimental_doubledouble",
srcs = ["experimental/doubledouble.py"],
srcs_version = "PY3",
deps = [":jax"],
)
pytype_library(
name = "experimental_host_callback",
srcs = ["experimental/host_callback.py"],