MCPcopy Index your code

hub / github.com/apple/ml-aim / functions

Functions203 in github.com/apple/ml-aim

⨍ Functions 203◇ Types & classes 60
↓ 9 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
aim-v1/aim/v1/torch/layers.py:257
↓ 7 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
aim-v1/aim/v1/mlx/layers.py:197
↓ 5 callersFunction_aim
( img_size: Union[int, Tuple[int, int]], patch_size: Union[int, Tuple[int, int]], embed_dim: int,
aim-v1/aim/v1/torch/models.py:118
↓ 4 callersFunction_aim
( img_size: Union[int, Tuple[int, int]], patch_size: Union[int, Tuple[int, int]], embed_dim: int,
aim-v1/aim/v1/mlx/models.py:110
↓ 4 callersFunction_aim
( img_size: Union[int, Tuple[int, int]], patch_size: Union[int, Tuple[int, int]], embed_dim: int,
aim-v1/aim/v1/jax/models.py:100
↓ 4 callersMethodforward
( self, input_pixels: torch.Tensor, input_ids: torch.Tensor, output_features:
aim-v2/aim/v2/torch/models.py:181
↓ 3 callersMethod__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
aim-v2/aim/v2/mlx/models.py:142
↓ 3 callersFunction_get_model_func_and_img_size
(model_name: ModelName)
aim-v2/aim/v2/utils.py:48
↓ 3 callersMethodforward
(self, x: torch.Tensor)
aim-v1/aim/v1/torch/layers.py:274
↓ 3 callersMethodupdate
(self, **kwargs: Any)
aim-v1/aim/v1/logger.py:97
↓ 2 callersMethod__init__
(self, preprocessor: nn.Module, trunk: nn.Module, head: nn.Module)
aim-v1/aim/v1/torch/models.py:98
↓ 2 callersMethod__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
aim-v2/aim/v2/torch/models.py:127
↓ 2 callersMethod_get_1d_sincos_pos_embed_from_grid
( embed_dim: int, pos: torch.Tensor )
aim-v1/aim/v1/torch/layers.py:58
↓ 2 callersMethod_get_1d_sincos_pos_embed_from_grid
(embed_dim: int, pos: mx.array)
aim-v1/aim/v1/mlx/layers.py:47
↓ 2 callersMethod_get_1d_sincos_pos_embed_from_grid
(embed_dim: int, pos: jax.Array)
aim-v1/aim/v1/jax/layers.py:85
↓ 2 callersFunctionprint
(*args, **kwargs)
aim-v1/aim/v1/utils.py:100
↓ 2 callersMethodupdate
(self, value: float, n: int = 1)
aim-v1/aim/v1/logger.py:44
↓ 1 callersMethod__init__
( self, preprocessor: nn.Module, trunk: nn.Module, head: nn.Module, )
aim-v1/aim/v1/mlx/models.py:83
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: int, use_bias: bool = True,
aim-v2/aim/v2/torch/layers.py:69
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: int, use_bias: bool = True,
aim-v2/aim/v2/mlx/layers.py:64
↓ 1 callersFunction_get_attention_target
(dim: int, num_heads: int)
aim-v1/aim/v1/torch/models.py:111
↓ 1 callersFunction_get_attention_target
(dim: int, num_heads: int)
aim-v1/aim/v1/mlx/models.py:103
↓ 1 callersFunction_get_attention_target
(dim: int, num_heads: int)
aim-v1/aim/v1/jax/models.py:91
↓ 1 callersFunction_get_weights_fname
(backend: Literal["torch", "jax", "mlx"])
aim-v2/aim/v2/utils.py:36
↓ 1 callersFunction_load_jax
(model_name: ModelName, weights_path: str)
aim-v2/aim/v2/utils.py:76
↓ 1 callersFunction_load_mlx
(model_name: ModelName, weights_path: str)
aim-v2/aim/v2/utils.py:89
↓ 1 callersFunction_load_torch
(model_name: ModelName, weights_path: str)
aim-v2/aim/v2/utils.py:63
↓ 1 callersMethod_norm
(self, x: jax.Array)
aim-v2/aim/v2/jax/layers.py:79
↓ 1 callersFunction_train_transforms
()
aim-v1/aim/v1/torch/data.py:56
↓ 1 callersMethodencode_image
( self, input_pixels: torch.Tensor, mask: Optional[torch.Tensor] = None, outpu
aim-v2/aim/v2/torch/models.py:203
↓ 1 callersMethodencode_image
( self, input_pixels: mx.array, mask: Optional[mx.array] = None, output_featur
aim-v2/aim/v2/mlx/models.py:221
↓ 1 callersMethodencode_image
( self, input_pixels: jax.Array, mask: Optional[jax.Array] = None, output_feat
aim-v2/aim/v2/jax/models.py:223
↓ 1 callersMethodencode_text
( self, input_ids: torch.Tensor, mask: Optional[torch.Tensor] = None, output_f
aim-v2/aim/v2/torch/models.py:215
↓ 1 callersMethodencode_text
( self, input_ids: mx.array, mask: Optional[mx.array] = None, output_features:
aim-v2/aim/v2/mlx/models.py:233
↓ 1 callersMethodencode_text
( self, input_ids: jax.Array, mask: Optional[jax.Array] = None, output_feature
aim-v2/aim/v2/jax/models.py:235
↓ 1 callersFunctionevaluate
(model: nn.Module, data_loader: DataLoader)
aim-v1/main_attnprobe.py:19
↓ 1 callersMethodextract_features
( self, x: ArrayLike, mask: Optional[ArrayLike] = None, max_block_id: Optional
aim-v1/aim/v1/mixins.py:27
↓ 1 callersFunctionget_load_model_fn
(arch: str)
aim-v1/aim/v1/utils.py:207
↓ 1 callersFunctioninit_jax_params
(model: Any, seed: int = 0)
aim-v1/aim/v1/utils.py:160
↓ 1 callersMethodinitialize_weights
(self)
aim-v1/aim/v1/torch/layers.py:138
↓ 1 callersFunctionis_dist_avail_and_initialized
()
aim-v1/aim/v1/utils.py:57
↓ 1 callersFunctionload_state_dict
( loader: Callable[..., Dict[str, torch.Tensor]], backbone_loc: str, head_loc: Optiona
aim-v1/aim/v1/utils.py:224
↓ 1 callersMethodlog_every
( self, iterable: Collection[Any], print_freq: int, header: str = "",
aim-v1/aim/v1/logger.py:127
↓ 1 callersFunctionmain
(args: argparse.Namespace)
aim-v1/main_attnprobe.py:51
↓ 1 callersFunctionmerge_state_dicts
( backbone_state_dict: Dict[str, torch.Tensor], head_state_dict: Dict[str, torch.Tensor], *, a
aim-v1/aim/v1/utils.py:41
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process.
aim-v1/aim/v1/utils.py:94
↓ 1 callersMethodsynchronize_between_processes
(self)
aim-v1/aim/v1/logger.py:120
↓ 1 callersFunctiontorch_weights_to_jax
(state_dict: Dict[str, Any])
aim-v1/aim/v1/utils.py:126
↓ 1 callersFunctiontorch_weights_to_mlx
(state_dict: Dict[str, torch.Tensor])
aim-v1/aim/v1/utils.py:108
↓ 1 callersFunctionval_transforms
Validation transformations. Args: img_size: Image size. Returns: The transformation.
aim-v1/aim/v1/torch/data.py:60
Method__call__
(self, h: int, w: int, embed_dim: int)
aim-v1/aim/v1/mlx/layers.py:30
Method__call__
(self, x: mx.array)
aim-v1/aim/v1/mlx/layers.py:91
Method__call__
(self, x: mx.array, mask: Optional[mx.array] = None)
aim-v1/aim/v1/mlx/layers.py:123
Method__call__
(self, x: mx.array, mask: Optional[mx.array] = None)
aim-v1/aim/v1/mlx/layers.py:172
Method__call__
(self, x: mx.array)
aim-v1/aim/v1/mlx/layers.py:214
Method__call__
(self, x: mx.array, mask: Optional[mx.array] = None)
aim-v1/aim/v1/mlx/layers.py:252
Method__call__
(self, _: mx.array, layer_features: List[mx.array])
aim-v1/aim/v1/mlx/layers.py:265
Method__call__
(self, x: mx.array)
aim-v1/aim/v1/mlx/layers.py:308
Method__call__
( self, tokens: mx.array, mask: Optional[mx.array] = None, max_block_id: Optio
aim-v1/aim/v1/mlx/models.py:48
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, max_block_id: Optional[i
aim-v1/aim/v1/mlx/models.py:94
Method__call__
(self, x: jax.Array)
aim-v1/aim/v1/jax/layers.py:40
Method__call__
(self, h: int, w: int, embed_dim: int)
aim-v1/aim/v1/jax/layers.py:68
Method__call__
(self, x: jax.Array, mask: Optional[jax.Array] = None)
aim-v1/aim/v1/jax/layers.py:107
Method__call__
( self, x: jax.Array, training: bool = False, mask: Optional[jax.Array] = None )
aim-v1/aim/v1/jax/layers.py:159
Method__call__
(self, x: jax.Array, training: bool = False)
aim-v1/aim/v1/jax/layers.py:196
Method__call__
(self, x: jax.Array, mask: Optional[jax.Array] = None)
aim-v1/aim/v1/jax/layers.py:220
Method__call__
(self, _: jax.Array, layer_features: List[jax.Array])
aim-v1/aim/v1/jax/layers.py:243
Method__call__
(self, x: jax.Array)
aim-v1/aim/v1/jax/layers.py:265
Method__call__
( self, tokens: jax.Array, mask: Optional[jax.Array] = None, max_block_id: Opt
aim-v1/aim/v1/jax/models.py:28
Method__call__
( self, x: jax.Array, mask: Optional[jax.Array] = None, max_block_id: Optional
aim-v1/aim/v1/jax/models.py:82
Method__call__
(self, input_ids: mx.array)
aim-v2/aim/v2/mlx/layers.py:45
Method__call__
(self, tokens: mx.array, eos_token_mask: mx.array)
aim-v2/aim/v2/mlx/layers.py:55
Method__call__
(self, x: mx.array)
aim-v2/aim/v2/mlx/layers.py:78
Method__call__
( self, input_pixels: mx.array, mask: Optional[mx.array] = None, output_featur
aim-v2/aim/v2/mlx/models.py:87
Method__call__
( self, input_ids: mx.array, mask: Optional[mx.array] = None, output_features:
aim-v2/aim/v2/mlx/models.py:132
Method__call__
( self, input_pixels: mx.array, input_ids: mx.array, output_features: bool = F
aim-v2/aim/v2/mlx/models.py:194
Method__call__
(self, x: jax.Array)
aim-v2/aim/v2/jax/layers.py:31
Method__call__
(self, input_ids: jax.Array)
aim-v2/aim/v2/jax/layers.py:42
Method__call__
(self, tokens: jax.Array, eos_token_mask: jax.Array)
aim-v2/aim/v2/jax/layers.py:61
Method__call__
(self, x: jax.Array)
aim-v2/aim/v2/jax/layers.py:72
Method__call__
(self, x: jax.Array)
aim-v2/aim/v2/jax/layers.py:93
Method__call__
( self, input_pixels: jax.Array, mask: Optional[jax.Array] = None, output_feat
aim-v2/aim/v2/jax/models.py:87
Method__call__
( self, input_ids: jax.Array, mask: Optional[jax.Array] = None, output_feature
aim-v2/aim/v2/jax/models.py:130
Method__call__
( self, input_pixels: jax.Array, input_ids: jax.Array, output_features: bool =
aim-v2/aim/v2/jax/models.py:196
Method__getattr__
(self, attr: str)
aim-v1/aim/v1/logger.py:104
Method__init__
( self, window_size: int = 20, fmt: str = "{median:.4f} ({global_avg:.4f})" )
aim-v1/aim/v1/logger.py:36
Method__init__
(self, delimiter: str = "\t")
aim-v1/aim/v1/logger.py:93
Method__init__
(self, cls_token: bool = False)
aim-v1/aim/v1/torch/layers.py:35
Method__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
aim-v1/aim/v1/torch/layers.py:76
Method__init__
( self, patchifier: PatchEmbed, drop_patches: bool = False, cls_token: bool =
aim-v1/aim/v1/torch/layers.py:111
Method__init__
( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, attn_dro
aim-v1/aim/v1/torch/layers.py:178
Method__init__
(self, *args: Any, num_patches: int = 256, **kwargs: Any)
aim-v1/aim/v1/torch/layers.py:218
Method__init__
(self, dim: int, *args: Any, lora_rank: int = 8, **kwargs: Any)
aim-v1/aim/v1/torch/layers.py:240
Method__init__
( self, dim: int, attn_target: Callable[[bool], nn.Module], ffn_target: Callab
aim-v1/aim/v1/torch/layers.py:284
Method__init__
(self, layers: Sequence[int], reduce: bool = False)
aim-v1/aim/v1/torch/layers.py:325
Method__init__
( self, dim: int, out_features: int, num_heads: int = 12, num_queries:
aim-v1/aim/v1/torch/layers.py:344
Method__init__
( self, attn_target: Callable[[bool], nn.Module], embed_dim: int, num_blocks:
aim-v1/aim/v1/torch/models.py:28
Method__init__
(self, config: Dict[str, Any])
aim-v1/aim/v1/torch/models.py:106
Method__init__
(self, cls_token: bool = False)
aim-v1/aim/v1/mlx/layers.py:26
Method__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
aim-v1/aim/v1/mlx/layers.py:63
next →1–100 of 203, ranked by callers