python_library(
    name = "jit",
    srcs = glob([
        "*.py",
        "templates/*",
    ]),
    base_module = "tools.jit",
    visibility = ["PUBLIC"],
    deps = [
        "//torchgen:torchgen",
    ],
)