Functions — github.com/cactus-compute/needle

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

Functions285 in github.com/cactus-compute/needle

↓ 1 callersFunction_install_xla_log_filter

Drop XLA Triton autotuner noise from stderr. XLA's Triton GEMM autotuner logs failed candidate fusions via LOG(ERROR) in xtile_compiler.cc an

needle/cli.py:18

↓ 1 callersFunction_is_musl

()

needle/agent/fetch.py:19

↓ 1 callersFunction_is_optional

(annotation)

needle/agent/tools.py:51

↓ 1 callersMethod_json_body

(self)

needle/playground/server.py:110

↓ 1 callersFunction_json_type

(annotation)

needle/agent/tools.py:56

↓ 1 callersFunction_layer

(params, i)

needle/model/decode.py:46

↓ 1 callersFunction_lib_name

()

needle/agent/fetch.py:11

↓ 1 callersFunction_library_path

()

needle/__init__.py:12

↓ 1 callersFunction_lloyd_max_gaussian

(bits, iters=200, samples=400000, seed=0)

needle/model/quantize.py:95

↓ 1 callersFunction_log

(msg)

needle/playground/server.py:60

↓ 1 callersFunction_make_rollout

(cfg, max_len, scan_len)

needle/model/decode.py:307

↓ 1 callersFunction_mask_diag

(mask, offset)

needle/model/architecture.py:133

↓ 1 callersFunction_mhc

(params, cfg)

needle/model/decode.py:186

↓ 1 callersFunction_nearest_idx

(x, cb)

needle/model/export.py:135

↓ 1 callersFunction_openrouter

(messages, model, api_key, temperature=0.9)

needle/model/finetune.py:49

↓ 1 callersFunction_pack_ternary_crumbs

(idx)

needle/model/export.py:174

↓ 1 callersFunction_packed_row_bytes

(in_pad, bits, group)

needle/model/export.py:168

↓ 1 callersFunction_parse_doc

(doc)

needle/agent/tools.py:94

↓ 1 callersFunction_platform_tag

()

needle/agent/fetch.py:32

↓ 1 callersFunction_q

(name, mat, bits, group)

needle/model/export.py:232

↓ 1 callersMethod_resolve

(self, tools)

needle/__init__.py:78

↓ 1 callersFunction_rollout_fn

(cfg, max_len, scan_len)

needle/model/decode.py:342

↓ 1 callersFunction_shard

(x, nd)

needle/model/decode.py:294

↓ 1 callersFunction_step_stats

(step_logits)

needle/model/run.py:110

↓ 1 callersFunction_submit

()

needle/model/finetune.py:104

↓ 1 callersFunction_tensors

(params, config, bits, group)

needle/model/export.py:240

↓ 1 callersFunction_tokenizer_blob

(tok)

needle/model/export.py:317

↓ 1 callersFunction_unpack_ternary_crumbs

(packed, in_pad)

needle/model/export.py:179

↓ 1 callersFunction_updateSteps

(current)

needle/playground/app.js:233

↓ 1 callersFunctionadd

(dst, src, scale)

needle/model/run.py:21

↓ 1 callersFunctionaddTurn

(query, data)

needle/playground/app.js:58

↓ 1 callersFunctionapplyPreset

(key)

needle/playground/app.js:146

↓ 1 callersFunctioncloseModal

(e)

needle/playground/app.js:182

↓ 1 callersMethodcomplete

(self, tools_json, query)

needle/playground/server.py:30

↓ 1 callersFunctioncq_distortion

(bits, group_size=CQ_GROUP_SIZE, rows=2048, seed=0)

needle/model/quantize.py:225

↓ 1 callersFunctioncq_fake_quant_kv

(x, bits, group=64)

needle/model/quantize.py:37

↓ 1 callersFunctioncq_mixed_params

(params, bits_map, default_bits, group_size=CQ_GROUP_SIZE)

needle/model/quantize.py:321

↓ 1 callersFunctioncq_quantize_params

(params, bits, group_size=128)

needle/model/quantize.py:161

↓ 1 callersFunctionemit

(msg)

needle/model/finetune.py:276

↓ 1 callersFunctiongenerate

(model, params, tokenizer, prompt, max_new_tokens=256, temperature=0.0, seed=0, stream=True)

needle/model/run.py:176

↓ 1 callersFunctiongenerate_main

(args)

needle/model/finetune.py:167

↓ 1 callersFunctiongreet

Greet someone.

tests/test_tools.py:135

↓ 1 callersMethodhas_default

(self)

needle/agent/tools.py:32

↓ 1 callersFunctionhead_dims

(config)

needle/model/architecture.py:14

↓ 1 callersFunctioninit_kv_cache

(config, batch, max_len)

needle/model/decode.py:34

↓ 1 callersFunctionkv_budget_window

(config)

needle/model/architecture.py:600

↓ 1 callersFunctionleaf_name

(path)

needle/model/quantize.py:285

↓ 1 callersFunctionmake_padding_mask

(tokens, pad_token_id)

needle/model/architecture.py:590

↓ 1 callersFunctionparse_tokenizer_blob

(blob)

needle/model/export.py:443

↓ 1 callersFunctionquantize_params

(params, group_size=128, bits=4)

needle/model/quantize.py:25

↓ 1 callersMethodreset

(self)

needle/playground/server.py:40

↓ 1 callersFunctionrollout

(params, buf, valid, sink, cos, sin, rng, temperature)

needle/model/decode.py:310

↓ 1 callersMethodrun

(self, query, max_steps=8, max_new_tokens=256)

needle/__init__.py:99

↓ 1 callersFunctionsend

()

needle/playground/app.js:108

↓ 1 callersFunctionto_fp16

(t)

needle/model/run.py:40

↓ 1 callersFunctiontrain_step

(lora, opt_state, ids, mask)

needle/model/finetune.py:311

Method__call__

(self, x)

needle/model/architecture.py:51

Method__call__

(self, indices, ngram_ok, tap_ok, quant=False)

needle/model/architecture.py:189

Method__call__

(self, x, mask=None, rope=None, quant=False)

needle/model/architecture.py:217

Method__call__

(self, x)

needle/model/architecture.py:290

Method__call__

(self, x, mask=None, rope=None, quant=False, engram_kv=None, site_flags=None)

needle/model/architecture.py:316

Method__call__

(self, x, xs, mask, rope, quant, engram_kv)

needle/model/architecture.py:349

Method__call__

(self, x, mask=None, rope=None, engram_kv=None, collect_hidden=False, quant=False)

needle/model/architecture.py:378

Method__call__

(self, cells, keep=None)

needle/model/architecture.py:449

Method__call__

(self, cells, keep=None)

needle/model/architecture.py:466

Method__call__

(self, tokens, mask=None, quant=False, return_mtp=False)

needle/model/architecture.py:517

Method__call__

(self, texts, truncation=True, max_length=None, **kwargs)

needle/model/tokenizer.py:72

Method__init__

(self, tools=None, system=None, weights=None, tool_index_path=None, buffer_size=65536)

needle/__init__.py:49

Method__init__

(self, **kwargs)

needle/model/architecture.py:82

Method__init__

(self, model_path)

needle/model/tokenizer.py:37

Method__init__

(self, name, dtype, shape, blob, group=0, bits=0)

needle/model/export.py:222

Method__init__

(self, meta)

needle/model/export.py:461

Method__init__

(self, weights=None)

needle/playground/server.py:18

Method__init__

(self, default=_MISSING, *, description=None, enum=None, const=_MISSING, ge=None, le=None, gt

needle/agent/tools.py:18

Function_conv_identity_init

(key, shape, dtype=jnp.float32)

needle/model/architecture.py:163

Function_finetune_worker

(tools_json, api_key, samples, engine)

needle/playground/server.py:66

Function_jsonable

(value)

needle/__init__.py:129

Function_res_identity_init

(key, shape, dtype=jnp.float32)

needle/model/architecture.py:175

Functionadd

Add two numbers.

tests/test_tools.py:15

Functionaverage_checkpoints

(paths, out)

needle/model/run.py:18

Functionbatch_generate

(model, params, tokenizer, prompts, max_new_tokens=96, return_signals=False)

needle/model/run.py:118

Functionbatched_generate

(config, params, tokenizer, prompts, max_new_tokens=96, temperature=0.0, seed=0, pad_to=N

needle/model/decode.py:350

Functionbody

(carry, _)

needle/model/decode.py:321

Methodbos_token_id

(self)

needle/model/tokenizer.py:57

Functionbuild_export

(params, config, bits=4, group=128, tokenizer=None, bits_map=None, kv_window=0)

needle/model/export.py:381

Functionconfigure_deploy

(act_bits=8, kv_bits=8, kv_group=64)

needle/model/quantize.py:61

Functionconfigure_qat

(every, weight_group=128, weight_bits=4)

needle/model/quantize.py:56

Functioncq_mixed_stats

(params, bits_map, default_bits, group_size=CQ_GROUP_SIZE)

needle/model/quantize.py:331

Functioncq_model_bytes

(params, bits, group_size=128)

needle/model/quantize.py:175

Functioncq_ste_mixed_params

(params, bits_map, default_bits, group_size=CQ_GROUP_SIZE)

needle/model/quantize.py:362

Functioncq_ste_params

(params, bits, group_size=CQ_GROUP_SIZE)

needle/model/quantize.py:358

Methoddecode

(self, ids)

needle/model/tokenizer.py:67

Functiondeploy_quantize

(params, config)

needle/model/quantize.py:74

FunctiondismissToast

()

needle/playground/app.js:13

Methoddo_GET

(self)

needle/playground/server.py:114

Methoddo_POST

(self)

needle/playground/server.py:133

Methodencode

(self, text)

needle/model/tokenizer.py:64

Methodencode_contrastive

(self, tokens, quant=False, window=0, sink=None)

needle/model/architecture.py:560

Methodeos_token_id

(self)

needle/model/tokenizer.py:53

Methodextract

(self, text, schema, max_new_tokens=256)

needle/__init__.py:121

← previous next →101–200 of 285, ranked by callers