Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/google-deepmind/penzai
/ functions
Functions
1,092 in github.com/google-deepmind/penzai
⨍
Functions
1,092
◇
Types & classes
360
↳
Endpoints
3
Method
test_einsum_verbose
(self)
tests/deprecated/v1/nn/linear_and_affine_test.py:32
Method
test_elementwise
(self)
tests/nn/basic_ops_test.py:24
Method
test_elementwise
(self)
tests/deprecated/v1/nn/basic_ops_test.py:25
Method
test_embedding_decode
(self)
tests/deprecated/v1/nn/embedding_test.py:32
Method
test_embedding_lookup
(self)
tests/deprecated/v1/nn/embedding_test.py:24
Method
test_embeddings
(self)
tests/nn/embedding_test.py:24
Method
test_freeze_all_variables
(self)
tests/core/variables_test.py:215
Method
test_get_and_substitute_dimension_variables
(self)
tests/core/shapecheck_test.py:698
Method
test_gpt_neox_consistency
(self)
tests/models/transformer_consistency_test.py:111
Method
test_gradient_checkpointing
(self)
tests/toolshed/gradient_checkpointing_test.py:28
Method
test_group_call_runs_children
(self, which)
tests/nn/grouping_test.py:34
Method
test_group_call_runs_children
(self, which)
tests/deprecated/v1/nn/grouping_test.py:34
Method
test_hoist_constant_side_inputs
(self)
tests/deprecated/v1/data_effects/side_input_test.py:118
Method
test_hoist_shared_state_requests
(self)
tests/deprecated/v1/data_effects/local_state_test.py:305
Method
test_inconsistent_overlap_vars_for_axes_and_unpack
(self)
tests/core/shapecheck_test.py:594
Method
test_indexed_update_at
(self)
tests/core/named_axes_test.py:879
Method
test_indexing_batched
(self)
tests/core/named_axes_test.py:445
Method
test_indexing_by_dict
(self)
tests/core/named_axes_test.py:413
Method
test_indexing_positional
(self)
tests/core/named_axes_test.py:396
Method
test_initialize_parameters_detects_duplicate_names
(self)
tests/deprecated/v1/nn/parameters_test.py:209
Method
test_initialize_parameters_sharded
(self)
tests/deprecated/v1/toolshed/sharding_util_test.py:283
Method
test_initialize_parameters_success
(self)
tests/deprecated/v1/nn/parameters_test.py:178
Method
test_inline_anonymous_sequentials
(self)
tests/nn/grouping_test.py:50
Method
test_inline_anonymous_sequentials
(self)
tests/deprecated/v1/nn/grouping_test.py:50
Method
test_invalid_namedarray
(self)
tests/core/named_axes_test.py:198
Method
test_invalid_namedarrayview
(self)
tests/core/named_axes_test.py:232
Method
test_invert__example_1
(self)
tests/core/selectors_test.py:569
Method
test_invert__example_2
(self)
tests/core/selectors_test.py:591
Method
test_jit_wrapper
(self)
tests/toolshed/jit_wrapper_test.py:28
Method
test_knock_out_heads
(self)
tests/toolshed/model_rewiring_test.py:26
Method
test_knock_out_heads
(self)
tests/deprecated/v1/toolshed/model_rewiring_test.py:26
Method
test_layer_call_input_structure_decorator_required
(self)
tests/deprecated/v1/shapecheck_layer_test.py:72
Method
test_layer_call_opt_out_decorator
(self)
tests/deprecated/v1/shapecheck_layer_test.py:106
Method
test_layer_call_output_structure_decorator_required
(self)
tests/deprecated/v1/shapecheck_layer_test.py:89
Method
test_layer_stack_build
(self)
tests/nn/layer_stack_test.py:111
Method
test_layer_variable_call
(self)
tests/nn/layer_test.py:23
Method
test_layernorm
(self)
tests/nn/standardization_test.py:38
Method
test_layernorm
(self)
tests/deprecated/v1/nn/standardization_test.py:32
Method
test_linear_in_place
(self)
tests/nn/linear_and_affine_test.py:111
Method
test_linear_in_place
(self)
tests/deprecated/v1/nn/linear_and_affine_test.py:69
Method
test_linear_not_in_place
(self)
tests/nn/linear_and_affine_test.py:91
Method
test_linear_not_in_place
(self)
tests/deprecated/v1/nn/linear_and_affine_test.py:59
Method
test_linearize_and_adjust
(self)
tests/toolshed/model_rewiring_test.py:118
Method
test_linearize_and_adjust
(self)
tests/deprecated/v1/toolshed/model_rewiring_test.py:118
Method
test_llama_consistency
(self, num_attention_heads, num_key_value_heads)
tests/models/transformer_consistency_test.py:37
Method
test_make_parameter_keyed
(self)
tests/nn/parameters_test.py:28
Method
test_make_parameter_slot
(self)
tests/nn/parameters_test.py:52
Method
test_mismached_positional_dimension_vars
(self)
tests/core/shapecheck_test.py:186
Method
test_mismatched_vars_for_axes
(self)
tests/core/shapecheck_test.py:531
Method
test_mistral_consistency
(self, num_attention_heads, num_key_value_heads)
tests/models/transformer_consistency_test.py:77
Method
test_multi_match_consistent
(self)
tests/core/shapecheck_test.py:667
Method
test_multi_match_inconsistent
(self)
tests/core/shapecheck_test.py:642
Method
test_multiple_unpack_iterative_solve
(self)
tests/core/shapecheck_test.py:223
Method
test_name_to_name_sharding
(self)
tests/deprecated/v1/toolshed/sharding_util_test.py:31
Method
test_named_array_equality
(self)
tests/core/named_axes_test.py:664
Method
test_named_array_tag
(self)
tests/core/named_axes_test.py:45
Method
test_named_array_untag
(self)
tests/core/named_axes_test.py:71
Method
test_named_array_untag_nothing
(self)
tests/core/named_axes_test.py:87
Method
test_named_array_wrap_named
(self)
tests/core/named_axes_test.py:58
Method
test_named_axis_dimension_variables
(self)
tests/core/shapecheck_test.py:296
Method
test_named_axis_inconsistent_shapes
(self)
tests/core/shapecheck_test.py:403
Method
test_named_multiple_unpack_iterative_solve
(self)
tests/core/shapecheck_test.py:448
Method
test_named_unpack_stuck
(self)
tests/core/shapecheck_test.py:474
Method
test_named_unpack_substitution_conflict
(self)
tests/core/shapecheck_test.py:364
Method
test_nmap
(self)
tests/core/named_axes_test.py:322
Method
test_object_mismatches
(self)
tests/core/shapecheck_test.py:39
Method
test_object_rendering
( self, *, target: Any = None, target_builder: Callable[[], Any] | None = None,
tests/treescope/renderer_test.py:246
Method
test_order_like
(self, left_structure, right_structure)
tests/core/named_axes_test.py:785
Method
test_ordered_as
(self)
tests/core/named_axes_test.py:177
Method
test_parameter_prefix
(self)
tests/deprecated/v1/nn/parameters_test.py:139
Method
test_partition_by_selection
(self)
tests/core/partitioning_test.py:23
Method
test_penzai_named_axes_info
(self)
tests/treescope/ndarray_adapters_test.py:162
Method
test_pick_nth_selected
(self)
tests/core/selectors_test.py:557
Method
test_positional_prefix
(self)
tests/core/named_axes_test.py:130
Method
test_positional_unpack_stuck
(self)
tests/core/shapecheck_test.py:244
Method
test_pytree_dataclass_field_inheritance_disallowed_by_default
(self)
tests/core/struct_pytree_dataclass_test.py:137
Method
test_pytree_dataclass_mutable_init_proxy_disabled
(self)
tests/core/struct_pytree_dataclass_test.py:238
Method
test_pytree_dataclass_mutable_init_proxy_enabled
(self)
tests/core/struct_pytree_dataclass_test.py:193
Method
test_pytree_dataclass_overwrite_parent_init
(self)
tests/core/struct_pytree_dataclass_test.py:262
Method
test_pytree_dataclass_overwrite_parent_warning_ignores_generated
(self)
tests/core/struct_pytree_dataclass_test.py:325
Method
test_pytree_dataclass_properties
(self)
tests/core/struct_pytree_dataclass_test.py:31
Method
test_pytree_dataclass_requires_pytree_methods
(self)
tests/core/struct_pytree_dataclass_test.py:100
Method
test_random_basic
(self)
tests/deprecated/v1/data_effects/random_test.py:55
Method
test_random_frozen
(self)
tests/deprecated/v1/data_effects/random_test.py:113
Method
test_random_predicate
(self)
tests/deprecated/v1/data_effects/random_test.py:60
Method
test_random_stream_lifecycle
(self)
tests/deprecated/v1/misc_util_test.py:25
Method
test_random_stream_mark_active
(self)
tests/deprecated/v1/misc_util_test.py:50
Method
test_random_with_state
(self)
tests/deprecated/v1/data_effects/random_test.py:97
Method
test_refine
(self)
tests/core/selectors_test.py:612
Method
test_rename_axes_multi
(self)
tests/nn/linear_and_affine_test.py:32
Method
test_rename_axes_multi
(self)
tests/deprecated/v1/nn/linear_and_affine_test.py:28
Method
test_rename_axes_one
(self)
tests/nn/linear_and_affine_test.py:25
Method
test_rename_axes_one
(self)
tests/deprecated/v1/nn/linear_and_affine_test.py:24
Method
test_residual
(self)
tests/nn/linear_and_affine_test.py:171
Method
test_residual
(self)
tests/deprecated/v1/nn/linear_and_affine_test.py:103
Method
test_rewire_computation_paths
(self)
tests/toolshed/model_rewiring_test.py:57
Method
test_rewire_computation_paths
(self)
tests/deprecated/v1/toolshed/model_rewiring_test.py:57
Method
test_rngs
(self)
tests/toolshed/unflaxify_test.py:151
Method
test_same_empty
(self)
tests/core/shapecheck_test.py:28
Method
test_same_objects
(self)
tests/core/shapecheck_test.py:32
← previous
next →
801–900 of 1,092, ranked by callers