MCPcopy Create free account

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

Functions203 in github.com/apple/ml-aim

⨍ Functions 203◇ Types & classes 60
Method__init__
( self, patchifier: PatchEmbed, drop_patches: bool = False, cls_token: bool =
aim-v1/aim/v1/mlx/layers.py:99
Method__init__
( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, attn_dro
aim-v1/aim/v1/mlx/layers.py:152
Method__init__
( self, dim: int, attn_target: Callable[[bool], nn.Module], ffn_target: Callab
aim-v1/aim/v1/mlx/layers.py:224
Method__init__
(self, layers: Sequence[int], reduce: bool = False)
aim-v1/aim/v1/mlx/layers.py:260
Method__init__
( self, dim: int, out_features: int, num_heads: int = 12, num_queries:
aim-v1/aim/v1/mlx/layers.py:277
Method__init__
( self, attn_target: Callable[[bool], nn.Module], embed_dim: int, num_blocks:
aim-v1/aim/v1/mlx/models.py:15
Method__init__
( self, vocab_size: int, embed_dim: int, max_context_length: int = 77,
aim-v2/aim/v2/torch/layers.py:31
Method__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
aim-v2/aim/v2/torch/models.py:31
Method__init__
( self, embed_dim: int = 768, mlp_hidden_dim: int = 2048, num_blocks: int = 12
aim-v2/aim/v2/torch/models.py:90
Method__init__
( self, vocab_size: int, embed_dim: int, max_context_length: int = 77,
aim-v2/aim/v2/mlx/layers.py:30
Method__init__
( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,
aim-v2/aim/v2/mlx/models.py:30
Method__init__
( self, embed_dim: int = 768, mlp_hidden_dim: int = 2048, num_blocks: int = 12
aim-v2/aim/v2/mlx/models.py:97
Method__post_init__
(self)
aim-v1/aim/v1/jax/layers.py:32
Method__str__
(self)
aim-v1/aim/v1/logger.py:82
Method__str__
(self)
aim-v1/aim/v1/logger.py:114
Functionaccuracy
( output: torch.Tensor, target: torch.Tensor, topk: Tuple[int, ...] = (1,) )
aim-v1/aim/v1/utils.py:27
Methodadd_meter
(self, name: str, meter: SmoothedValue)
aim-v1/aim/v1/logger.py:124
Functionaim_1B
(*args: Any, **kwargs: Any)
aim-v1/hubconf.py:16
Functionaim_1B
(img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any)
aim-v1/aim/v1/torch/models.py:184
Functionaim_1B
(img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any)
aim-v1/aim/v1/mlx/models.py:175
Functionaim_1B
(img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any)
aim-v1/aim/v1/jax/models.py:165
Functionaim_3B
(*args: Any, **kwargs: Any)
aim-v1/hubconf.py:20
Functionaim_3B
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: int = 14, **kwargs: Any )
aim-v1/aim/v1/torch/models.py:196
Functionaim_3B
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: int = 14, **kwargs: Any )
aim-v1/aim/v1/mlx/models.py:186
Functionaim_3B
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: int = 14, **kwargs: Any )
aim-v1/aim/v1/jax/models.py:176
Functionaim_600M
(*args: Any, **kwargs: Any)
aim-v1/hubconf.py:12
Functionaim_600M
(img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any)
aim-v1/aim/v1/torch/models.py:172
Functionaim_600M
(img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any)
aim-v1/aim/v1/mlx/models.py:164
Functionaim_600M
(img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any)
aim-v1/aim/v1/jax/models.py:154
Functionaim_7B
(*args: Any, **kwargs: Any)
aim-v1/hubconf.py:24
Functionaim_7B
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: int = 14, **kwargs: Any )
aim-v1/aim/v1/torch/models.py:210
Functionaim_7B
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: int = 14, **kwargs: Any )
aim-v1/aim/v1/mlx/models.py:199
Functionaim_7B
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: int = 14, **kwargs: Any )
aim-v1/aim/v1/jax/models.py:189
Functionaimv2_1B
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/torch/models.py:277
Functionaimv2_1B
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/mlx/models.py:295
Functionaimv2_1B
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/jax/models.py:297
Functionaimv2_3B
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/torch/models.py:294
Functionaimv2_3B
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/mlx/models.py:312
Functionaimv2_3B
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/jax/models.py:314
Functionaimv2_base
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/torch/models.py:226
Functionaimv2_base
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/mlx/models.py:244
Functionaimv2_base
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/jax/models.py:246
Functionaimv2_huge
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/torch/models.py:260
Functionaimv2_huge
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/mlx/models.py:278
Functionaimv2_huge
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/jax/models.py:280
Functionaimv2_large
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/torch/models.py:243
Functionaimv2_large
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/mlx/models.py:261
Functionaimv2_large
( img_size: Union[int, Tuple[int, int]] = 224, **kwargs: Any, )
aim-v2/aim/v2/jax/models.py:263
Functionaimv2_large_lit
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int, int]] = 14, vocab
aim-v2/aim/v2/torch/models.py:325
Functionaimv2_large_lit
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int, int]] = 14, vocab
aim-v2/aim/v2/mlx/models.py:343
Functionaimv2_large_lit
( img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int, int]] = 14, vocab
aim-v2/aim/v2/jax/models.py:345
Functionaimv2_large_native
(**kwargs: Any)
aim-v2/aim/v2/torch/models.py:311
Functionaimv2_large_native
(**kwargs: Any)
aim-v2/aim/v2/mlx/models.py:329
Functionaimv2_large_native
(**kwargs: Any)
aim-v2/aim/v2/jax/models.py:331
Methodavg
(self)
aim-v1/aim/v1/logger.py:66
Functioncallback
(use_bias: bool)
aim-v1/aim/v1/torch/models.py:112
Functioncallback
(use_bias: bool)
aim-v1/aim/v1/mlx/models.py:104
Functioncallback
(use_bias: bool, name: str)
aim-v1/aim/v1/jax/models.py:92
Functioncreate_dataloader
Create a dataloader from a directory. Args: root: Path to the root of the dataset. split: Whether to create train or validation d
aim-v1/aim/v1/torch/data.py:19
Methodforward
( self, x: ArrayLike, mask: Optional[ArrayLike] = None, max_block_id: Optional
aim-v1/aim/v1/mixins.py:16
Methodforward
(self, h: int, w: int, embed_dim: int)
aim-v1/aim/v1/torch/layers.py:39
Methodforward
(self, x: torch.Tensor)
aim-v1/aim/v1/torch/layers.py:104
Methodforward
( self, x: torch.Tensor, mask: Optional[torch.Tensor] = None )
aim-v1/aim/v1/torch/layers.py:148
Methodforward
( self, x: torch.Tensor, mask: Optional[torch.Tensor] = None, **_: Any )
aim-v1/aim/v1/torch/layers.py:197
Methodforward
( self, x: torch.Tensor, mask: Optional[torch.Tensor] = None, **kwargs: Any )
aim-v1/aim/v1/torch/layers.py:225
Methodforward
( self, x: torch.Tensor, mask: Optional[torch.Tensor] = None )
aim-v1/aim/v1/torch/layers.py:315
Methodforward
( self, _: torch.Tensor, layer_features: List[torch.Tensor] )
aim-v1/aim/v1/torch/layers.py:330
Methodforward
(self, x: torch.Tensor)
aim-v1/aim/v1/torch/layers.py:370
Methodforward
( self, tokens: torch.Tensor, mask: Optional[torch.Tensor] = None, max_block_i
aim-v1/aim/v1/torch/models.py:63
Methodforward
( self, input_pixels: ArrayLike, mask: Optional[ArrayLike] = None, output_feat
aim-v2/aim/v2/mixins.py:16
Methodforward
( self, input_ids: ArrayLike, mask: Optional[ArrayLike] = None, output_feature
aim-v2/aim/v2/mixins.py:33
Methodforward
(self, input_ids: torch.Tensor)
aim-v2/aim/v2/torch/layers.py:46
Methodforward
( self, tokens: torch.Tensor, eos_token_mask: torch.Tensor )
aim-v2/aim/v2/torch/layers.py:56
Methodforward
(self, x: torch.Tensor)
aim-v2/aim/v2/torch/layers.py:83
Methodglobal_avg
(self)
aim-v1/aim/v1/logger.py:71
Methodgrid_size
(self)
aim-v1/aim/v1/jax/layers.py:55
Functionimg
()
aim-v1/tests/conftest.py:9
Functionimg
()
aim-v2/tests/conftest.py:9
Functioninit_distributed_mode
(dist_url: str)
aim-v1/aim/v1/utils.py:67
Functioninput_ids
()
aim-v2/tests/conftest.py:15
Functionis_main_process
()
aim-v1/aim/v1/utils.py:61
Functionload_pretrained
Load an autoregressive image model (AIM). Args: arch: Model architecture. pretrained: Whether to also load the pretrained weights
aim-v1/aim/v1/utils.py:173
Functionload_pretrained
Load pre-trained AIMv2 model. Args: model_name: Name of the model. backend: Compute backend. kwargs: Keyword arguments fo
aim-v2/aim/v2/utils.py:100
Methodmax
(self)
aim-v1/aim/v1/logger.py:75
Methodmax_block_id
(self)
aim-v1/aim/v1/torch/layers.py:339
Methodmax_block_id
(self)
aim-v1/aim/v1/mlx/layers.py:272
Methodmax_block_id
(self)
aim-v1/aim/v1/jax/layers.py:250
Methodmedian
(self)
aim-v1/aim/v1/logger.py:61
Methodnum_patches
(self)
aim-v1/aim/v1/jax/layers.py:60
Methodsetup
(self)
aim-v2/aim/v2/jax/models.py:45
Methodsetup
(self)
aim-v2/aim/v2/jax/models.py:105
Methodsetup
(self)
aim-v2/aim/v2/jax/models.py:157
Functionsetup_logger
(name: str = "AIM", level: int = logging.INFO)
aim-v1/aim/v1/logger.py:23
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
aim-v1/aim/v1/logger.py:49
Methodtest_image_encoder
(self, model_name: str, img: np.ndarray)
aim-v2/tests/test_backend.py:71
Methodtest_image_encoders
(self, model_name: str, img: np.ndarray)
aim-v2/tests/test_backend.py:30
Functiontest_jax_backend
(model: str, img: np.ndarray)
aim-v1/tests/test_backend.py:45
Functiontest_jax_dtype
(dtype: jnp.dtype)
aim-v1/tests/test_backend.py:89
Methodtest_lit
(self, img: np.ndarray, input_ids: np.ndarray)
aim-v2/tests/test_backend.py:48
Methodtest_lit
(self, img: np.ndarray, input_ids: np.ndarray)
aim-v2/tests/test_backend.py:90
← previousnext →101–200 of 203, ranked by callers