Add copybara rules for importing experimental PRs. After importing you can run build_cleaner third_party/py/neural_tangents/... in the new copybara workspace to create BUILD targets. There's still a lot of stuff to improve (some changes suggested in #142, and here would be nice to run build_cleaner automatically), but this could be a bit helpful in the meantime.

PiperOrigin-RevId: 436878167

Author: romanngg

neural_tangents/_src/utils/typing.py

@@ -14,9 +14,8 @@
 
 """Common Type Definitions."""
 
-
 from typing import Any, Dict, Generator, List, Optional, Sequence, TYPE_CHECKING, Tuple, TypeVar, Union
-from typing_extensions import Protocol
+
 import jax.numpy as np
 from jax import random
 from .kernel import Kernel
@@ -50,7 +49,6 @@
 """
 T = TypeVar('T')
 
-
 if TYPE_CHECKING:
   NTTree = Union[List['NTTree[T]'], Tuple['NTTree[T]', ...], T]
   NTTrees = Union[List['NTTree[T]'], Tuple['NTTree[T]', ...]]