Search results

BUG: potential tracer leak in _jitted_nmapped_getitem?

Status: Open.

#117 In google-deepmind/penzai;

· amifalk opened on Apr 26, 2025

Conversion from pretrained HuggingFace models

Status: Open.

#115 In google-deepmind/penzai;

· ealt opened on Apr 22, 2025

strange import behavior of penzai.nn and pz

Status: Open.

#114 In google-deepmind/penzai;

· hrbigelow opened on Apr 17, 2025

Feature request: Support for quantization

feature-requestNew feature or requestNew feature or request

Status: Open.

#113 In google-deepmind/penzai;

· JEM-Mosig opened on Apr 17, 2025

Conversion of a LlamaForCausalLM does not support <code>use_cache</code> and <code>_name_or_path</code>

Status: Open.

#112 In google-deepmind/penzai;

· JEM-Mosig opened on Apr 15, 2025

Initializer for Penzai's Linear layer does not support <code>jax.nn.initializers</code>

Status: Open.

#107 In google-deepmind/penzai;

· Samarendra109 opened on Feb 7, 2025

FR: expose device/sharding argument for NamedArray.wrap

Status: Open.

#80 In google-deepmind/penzai;

· amifalk opened on Aug 8, 2024

FR: batched mapping for <code>named_axes.nmap</code>

feature-requestNew feature or requestNew feature or request

Status: Open.

#71 In google-deepmind/penzai;

· amifalk opened on Jul 26, 2024