Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/deepseek-ai/Engram
/ functions
Functions
19 in github.com/deepseek-ai/Engram
⨍
Functions
19
◇
Types & classes
8
↓ 3 callers
Method
__init__
(self, layer_id)
engram_demo_v1.py:327
↓ 1 callers
Method
_build_lookup_table
(self)
engram_demo_v1.py:84
↓ 1 callers
Method
_compress
(self, input_ids)
engram_demo_v1.py:112
↓ 1 callers
Method
_get_ngram_hashes
( self, input_ids: np.ndarray, layer_id: int, )
engram_demo_v1.py:262
↓ 1 callers
Method
calculate_vocab_size_across_layers
(self)
engram_demo_v1.py:235
↓ 1 callers
Function
find_next_prime
(start, seen_primes)
engram_demo_v1.py:181
↓ 1 callers
Method
hash
(self, input_ids)
engram_demo_v1.py:298
Method
__call__
(self, input_ids)
engram_demo_v1.py:120
Method
__init__
( self, tokenizer_name_or_path, )
engram_demo_v1.py:61
Method
__init__
( self, hidden_size: int, kernel_size: int = 4, dilation: int = 1,
engram_demo_v1.py:124
Method
__init__
( self, engram_vocab_size, max_ngram_size, n_embed_per_ngram, n_head_
engram_demo_v1.py:189
Method
__init__
(self, list_of_N: List[int], D: int)
engram_demo_v1.py:306
Method
__init__
(self,layer_id)
engram_demo_v1.py:381
Method
__len__
(self)
engram_demo_v1.py:81
Method
forward
Input: (B,L,HC_MULT,D) Output: (B,L,HC_MULT,D)
engram_demo_v1.py:156
Method
forward
(self, input_ids: torch.Tensor)
engram_demo_v1.py:320
Method
forward
hidden_states: [B, L, HC_MULT, D] input_ids: [B, L]
engram_demo_v1.py:358
Method
forward
(self,input_ids,hidden_states)
engram_demo_v1.py:389
Method
shift_k
(k: int)
engram_demo_v1.py:272