SYMBOL INDEX (67 symbols across 15 files)

FILE: src/data.py
  function load_wikitext2 (line 10) | def load_wikitext2() -> Tuple[DataLoader[Any], DataLoader[Any], DataLoad...

FILE: src/memory/__init__.py
  class Memory (line 8) | class Memory:
    method add (line 13) | def add(self, embeddings: npt.NDArray[Any]) -> None:
    method remove (line 16) | def remove(self, ids: npt.NDArray[Any]) -> None:
    method update (line 19) | def update(self, ids: npt.NDArray[Any], updated_embeddings: npt.NDArra...
    method search (line 23) | def search(self, query_vectors: npt.NDArray[Any], k: int = 10) -> Any:
    method refresh (line 26) | def refresh(self) -> None:

FILE: src/memory/associative.py
  class AssociativeMemory (line 11) | class AssociativeMemory(Memory):
    method __init__ (line 12) | def __init__(
    method _to_numpy (line 78) | def _to_numpy(self, data: Union[npt.NDArray[Any], torch.Tensor]) -> Un...
    method add (line 89) | def add(self, embeddings: Union[npt.NDArray[Any], torch.Tensor]) -> None:
    method remove (line 110) | def remove(self, ids: Union[npt.NDArray[Any], torch.Tensor]) -> None:
    method update (line 128) | def update(
    method search (line 149) | def search(self, query_vectors: Any, k: int = 10) -> Any:
    method age_memory (line 178) | def age_memory(self, decay_factor: float = 0.999) -> None:
    method forget_randomly (line 198) | def forget_randomly(self) -> None:
    method garbage_collect (line 213) | def garbage_collect(self, threshold: float = 1e-6) -> npt.NDArray[Any]:
    method get_all_embeddings (line 237) | def get_all_embeddings(self) -> npt.NDArray[Any]:
    method __getitem__ (line 248) | def __getitem__(self, index: int) -> Any:
    method size (line 259) | def size(self) -> int:

FILE: src/memory/batch_associative.py
  class BatchAssociativeMemory (line 16) | class BatchAssociativeMemory:
    method __init__ (line 17) | def __init__(
    method batch_add (line 36) | def batch_add(self, embeddings: Union[npt.NDArray[Any], "torch.Tensor"...
    method batch_remove (line 46) | def batch_remove(self, ids: Union[npt.NDArray[Any], "torch.Tensor"]) -...
    method batch_search (line 55) | def batch_search(

FILE: src/memory/multitoken_batch_associative.py
  class MultiTokenBatchAssociativeMemory (line 16) | class MultiTokenBatchAssociativeMemory:
    method __init__ (line 17) | def __init__(
    method batch_add (line 36) | def batch_add(self, embeddings: Union[npt.NDArray[Any], "torch.Tensor"...
    method batch_remove (line 46) | def batch_remove(self, ids: Union[npt.NDArray[Any], "torch.Tensor"]) -...
    method batch_search (line 55) | def batch_search(

FILE: src/models/gpt2_associative.py
  class VardaGPTAssociative (line 10) | class VardaGPTAssociative(nn.Module):
    method __init__ (line 11) | def __init__(
    method forward (line 69) | def forward(self, input_vectors: torch.Tensor) -> Any:

FILE: src/models/gpt2_working.py
  class VardaGPTWorking (line 10) | class VardaGPTWorking(nn.Module):
    method __init__ (line 11) | def __init__(
    method forward (line 52) | def forward(self, input_vectors: torch.Tensor, memory_input: Optional[...

FILE: src/train.py
  function train (line 19) | def train(

FILE: src/train_parallel.py
  function train (line 19) | def train(

FILE: test/memory/test_associative.py
  function embeddings (line 10) | def embeddings(request):
  function test_add_and_search (line 18) | def test_add_and_search(embeddings):
  function test_update (line 29) | def test_update(embeddings):
  function test_refresh_memory (line 44) | def test_refresh_memory(embeddings):
  function test_getitem (line 57) | def test_getitem(embeddings):

FILE: test/memory/test_associative_batch.py
  function batch_memory (line 14) | def batch_memory():
  function tensor_type (line 22) | def tensor_type(request):
  function create_tensor (line 26) | def create_tensor(tensor_type, data):
  function test_batch_add (line 33) | def test_batch_add(batch_memory, tensor_type):
  function test_batch_remove (line 48) | def test_batch_remove(batch_memory, tensor_type):
  function test_batch_search (line 67) | def test_batch_search(batch_memory, tensor_type):

FILE: test/memory/test_memory_features.py
  function embeddings (line 10) | def embeddings(request):
  function forgetful_memory (line 19) | def forgetful_memory():
  function memory (line 27) | def memory():
  function test_age_memory (line 33) | def test_age_memory(embeddings, forgetful_memory):
  function test_forget_randomly (line 46) | def test_forget_randomly(embeddings, forgetful_memory):
  function test_garbage_collect (line 62) | def test_garbage_collect(embeddings, memory):

FILE: test/memory/test_memory_types.py
  function index_type (line 10) | def index_type(request):
  function tensor_type (line 15) | def tensor_type(request):
  function to_tensor (line 19) | def to_tensor(tensor_type, array):
  function test_associative_memory_add_remove_update_search (line 28) | def test_associative_memory_add_remove_update_search(index_type, tensor_...

FILE: test/memory/test_multitoken.py
  function test_batch_add (line 7) | def test_batch_add():
  function test_batch_remove (line 22) | def test_batch_remove():
  function test_batch_search (line 40) | def test_batch_search():
  function test_torch_tensors (line 60) | def test_torch_tensors():

FILE: test/models/test_gpt2_associative.py
  function varda_gpt_associative (line 9) | def varda_gpt_associative():
  function test_initialization (line 13) | def test_initialization(varda_gpt_associative):
  function test_forward_pass_no_memory (line 21) | def test_forward_pass_no_memory(varda_gpt_associative):
  function test_forward_pass_with_memory (line 32) | def test_forward_pass_with_memory(varda_gpt_associative):