MCPcopy Create free account

hub / github.com/cactus-compute/needle / functions

Functions285 in github.com/cactus-compute/needle

⨍ Functions 285◇ Types & classes 20
↓ 13 callersMethodencode
(self, text)
needle/model/export.py:502
↓ 10 callersFunctionbuild_schema
(fn)
needle/agent/tools.py:110
↓ 8 callersMethodload
(self)
needle/playground/server.py:25
↓ 7 callersMethoddecode
(self, ids)
needle/model/export.py:520
↓ 6 callersFunctiondefault_init
()
needle/model/architecture.py:28
↓ 6 callersFunctionget_tokenizer
(vocab_size=None)
needle/model/tokenizer.py:101
↓ 6 callersFunctionshowError
(msg)
needle/playground/app.js:6
↓ 5 callersFunction_cq_codebook_np
(bits, group_size)
needle/model/quantize.py:113
↓ 5 callersFunction_is_pydantic_model
(obj)
needle/agent/tools.py:143
↓ 5 callersFunction_lib
()
needle/__init__.py:32
↓ 5 callersFunction_map_quant_leaves
(params, fn)
needle/model/quantize.py:254
↓ 5 callersFunctionbuild_main
(args)
needle/model/finetune.py:348
↓ 5 callersMethodcomplete
(self, text, max_new_tokens=256)
needle/__init__.py:93
↓ 5 callersFunctioncq_quantize
(w, bits, group_size=128, codebook=None)
needle/model/quantize.py:134
↓ 5 callersFunctionread_export
(path)
needle/model/export.py:396
↓ 4 callersFunction_attn_width
(config)
needle/model/decode.py:22
↓ 4 callersFunction_bits_for
(name, bits_map, default_bits)
needle/model/quantize.py:218
↓ 4 callersFunctionfinetune_local
(args, progress=None)
needle/model/finetune.py:269
↓ 4 callersFunctionload_checkpoint
(path, return_run=False)
needle/model/run.py:61
↓ 4 callersFunctionmerge_lora
(params, lora, scale)
needle/model/finetune.py:260
↓ 4 callersFunctionrender_example
(example)
needle/model/finetune.py:187
↓ 3 callersFunction_aq
(x, quant)
needle/model/architecture.py:22
↓ 3 callersMethod_bind
(self)
needle/__init__.py:59
↓ 3 callersFunction_cq_hadamard_np
(group_size)
needle/model/quantize.py:119
↓ 3 callersMethod_engram_kv
(self, tokens, mask, quant)
needle/model/architecture.py:506
↓ 3 callersFunction_rms_unit
(x, epsilon=1e-6)
needle/model/architecture.py:158
↓ 3 callersMethod_rope
(self, seq_len)
needle/model/architecture.py:501
↓ 3 callersFunction_shift_right
(x, offset)
needle/model/architecture.py:125
↓ 3 callersFunction_zcrms
(x, scale, eps=1e-6)
needle/model/decode.py:41
↓ 3 callersMethodapply
(self, schema)
needle/agent/tools.py:35
↓ 3 callersFunctioneffective_kv_window
(config)
needle/model/architecture.py:611
↓ 3 callersFunctionengram_geometry
(config)
needle/model/architecture.py:118
↓ 3 callersFunctionextract
One-shot structured extraction: declare `schema` as the only tool and return the parsed object (a Pydantic instance if `schema` is a model, else a
needle/__init__.py:137
↓ 3 callersFunctionfetchModelName
()
needle/playground/app.js:34
↓ 3 callersFunctiongenerate_dataset
(tools, num_samples, model=DEFAULT_MODEL, batch_size=25, api_key=None, workers=8, progres
needle/model/finetune.py:92
↓ 3 callersFunctioninit_lora
(params, paths, rank, key)
needle/model/finetune.py:242
↓ 3 callersFunctionlora_target_paths
(params)
needle/model/finetune.py:229
↓ 3 callersFunctionmake_causal_mask
(seq_len)
needle/model/architecture.py:585
↓ 3 callersFunctionnewChat
()
needle/playground/app.js:136
↓ 3 callersFunctionparse_bits_map
(spec)
needle/model/quantize.py:301
↓ 3 callersFunctionprecompute_rope_freqs
(head_dim, seq_len, theta=10000.0)
needle/model/architecture.py:95
↓ 3 callersFunctionpydantic_schema
(model)
needle/agent/tools.py:149
↓ 3 callersFunctionquant_leaf_names
(params)
needle/model/quantize.py:289
↓ 3 callersFunctionwrite_export
(params, config, path, bits=4, group=128, tokenizer=None, bits_map=None, kv_window=0)
needle/model/export.py:387
↓ 2 callersFunction_build_args
(checkpoint, out, bits="4", lora=None)
tests/test_build.py:9
↓ 2 callersFunction_collect_tools
(examples)
needle/model/finetune.py:140
↓ 2 callersFunction_doc_prefix_len
(ids)
needle/model/decode.py:12
↓ 2 callersFunction_encode
(tokenizer, example, max_len)
needle/model/finetune.py:202
↓ 2 callersMethod_encode_contrastive
(self, tokens, quant=False, window=0, sink=None)
needle/model/architecture.py:554
↓ 2 callersFunction_f32
(params)
needle/model/decode.py:250
↓ 2 callersFunction_finetune_args
(data, checkpoint, out, ckpt_dir)
tests/test_finetune.py:27
↓ 2 callersFunction_forward_cached
(params, cfg, tokens, k_cache, v_cache, pos, cos, sin, key_valid=None, flash=False, hist=N
needle/model/decode.py:198
↓ 2 callersFunction_fp16
(name, arr)
needle/model/export.py:227
↓ 2 callersFunction_geometry
(config)
needle/model/export.py:103
↓ 2 callersFunction_get_decode_fn
(model, buf_len)
needle/model/run.py:100
↓ 2 callersFunction_is_quant_leaf
(path, leaf)
needle/model/quantize.py:150
↓ 2 callersFunction_pack_cact
(params, config, bits, group, tokenizer, kv_window=0)
needle/model/export.py:340
↓ 2 callersFunction_pack_lsb
(idx, bits)
needle/model/export.py:143
↓ 2 callersFunction_parse_array
(text)
needle/model/finetune.py:62
↓ 2 callersFunction_reduces_second_last
(path)
needle/model/quantize.py:156
↓ 2 callersFunction_resetModal
()
needle/playground/app.js:188
↓ 2 callersFunction_sample
(last, rng, temperature)
needle/model/decode.py:298
↓ 2 callersMethod_send
(self, code, body, ctype="application/json")
needle/playground/server.py:102
↓ 2 callersFunction_sinkhorn
(logits, iters=20)
needle/model/architecture.py:167
↓ 2 callersFunction_unique_generator
()
tests/test_generate.py:59
↓ 2 callersFunction_unpack_lsb
(packed, bits, in_pad)
needle/model/export.py:155
↓ 2 callersFunction_walsh_matrix
(n)
needle/model/architecture.py:278
↓ 2 callersFunction_write_data
(path)
tests/test_finetune.py:14
↓ 2 callersFunctionadd_cq_noise
(w, key, scale, group_size=CQ_GROUP_SIZE)
needle/model/quantize.py:242
↓ 2 callersFunctionapply_rope
(x, cos, sin)
needle/model/architecture.py:102
↓ 2 callersFunctionaugment_jsonl
(path, num_samples, model=DEFAULT_MODEL, batch_size=25, out_path=None, workers=8)
needle/model/finetune.py:151
↓ 2 callersFunctioncanonical_tensor_name
(name)
needle/model/quantize.py:203
↓ 2 callersFunctioncq_ste
(w, bits, group_size=CQ_GROUP_SIZE)
needle/model/quantize.py:354
↓ 2 callersFunctiondecode_cfg
(config, kv_window=0)
needle/model/decode.py:26
↓ 2 callersFunctiondecode_fn
(params, tokens)
needle/model/run.py:104
↓ 2 callersFunctionengram_indices
(tokens, orders, heads, slots)
needle/model/architecture.py:144
↓ 2 callersFunctionfake_quant
(w, group_size=128, bits=4)
needle/model/quantize.py:10
↓ 2 callersFunctionfn
(w, i)
needle/model/quantize.py:324
↓ 2 callersMethodhidden_cells
(self, tokens, quant=False, window=0, sink=None)
needle/model/architecture.py:538
↓ 2 callersFunctionload_jsonl
(path, tokenizer, max_len)
needle/model/finetune.py:213
↓ 2 callersMethodload_weights
(self, path)
needle/playground/server.py:48
↓ 2 callersFunctionprobe_pool
(cells, probes, keep=None, dtype=jnp.bfloat16)
needle/model/architecture.py:426
↓ 2 callersFunctionresidual_init
(num_layers)
needle/model/architecture.py:32
↓ 1 callersFunction_align
(n)
needle/model/export.py:131
↓ 1 callersFunction_attn_cached
(x, lp, k_cache_l, v_cache_l, pos, cos_s, sin_s, cfg, key_valid, flash=False, sink=None)
needle/model/decode.py:63
↓ 1 callersFunction_block_cached
(x, lp, k_cache_l, v_cache_l, pos, cos_s, sin_s, cfg, key_valid, flash=False, sink=None)
needle/model/decode.py:137
↓ 1 callersMethod_bpe
(self, seg)
needle/model/export.py:480
↓ 1 callersFunction_cq_nearest
(x, cb)
needle/model/quantize.py:126
↓ 1 callersFunction_cq_pack
(w, bits, group)
needle/model/export.py:184
↓ 1 callersFunction_cq_unpack
(packed, norms, out, in_dim, bits, group)
needle/model/export.py:202
↓ 1 callersFunction_dedup_key
(example)
needle/model/finetune.py:87
↓ 1 callersFunction_download_tokenizer_from_hf
(prefix)
needle/model/tokenizer.py:82
↓ 1 callersFunction_engine_available
()
tests/conftest.py:7
↓ 1 callersFunction_engram_kv
(params, cfg, hist, hist_valid, pos, S)
needle/model/decode.py:156
↓ 1 callersFunction_engram_window
(cfg)
needle/model/decode.py:152
↓ 1 callersFunction_field_of
(annotation, default)
needle/agent/tools.py:84
↓ 1 callersMethod_gate
(self, name)
needle/model/architecture.py:312
↓ 1 callersFunction_get
(params, keys)
needle/model/export.py:310
↓ 1 callersFunction_hadamard
(z, d1, d2, d3, d_model)
needle/model/decode.py:126
↓ 1 callersFunction_head_tensors
(params)
needle/model/export.py:292
next →1–100 of 285, ranked by callers