Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/apple/ml-aim
/ types & classes
Types & classes
60 in github.com/apple/ml-aim
⨍
Functions
203
◇
Types & classes
60
↓ 7 callers
Class
AIMv2VisionEncoder
aim-v2/aim/v2/torch/models.py:30
↓ 7 callers
Class
AIMv2VisionEncoder
aim-v2/aim/v2/mlx/models.py:29
↓ 7 callers
Class
AIMv2VisionEncoder
aim-v2/aim/v2/jax/models.py:30
↓ 4 callers
Class
AIM
aim-v1/aim/v1/torch/models.py:97
↓ 2 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
aim-v1/aim/v1/logger.py:31
↓ 1 callers
Class
AIM
aim-v1/aim/v1/mlx/models.py:82
↓ 1 callers
Class
AIM
aim-v1/aim/v1/jax/models.py:76
↓ 1 callers
Class
AIMv2LiT
aim-v2/aim/v2/torch/models.py:126
↓ 1 callers
Class
AIMv2LiT
aim-v2/aim/v2/mlx/models.py:141
↓ 1 callers
Class
AIMv2LiT
aim-v2/aim/v2/jax/models.py:139
↓ 1 callers
Class
AIMv2TextEncoder
aim-v2/aim/v2/torch/models.py:89
↓ 1 callers
Class
AIMv2TextEncoder
aim-v2/aim/v2/mlx/models.py:96
↓ 1 callers
Class
AIMv2TextEncoder
aim-v2/aim/v2/jax/models.py:96
↓ 1 callers
Class
SinCosPosEmbed
aim-v1/aim/v1/torch/layers.py:34
↓ 1 callers
Class
SinCosPosEmbed
aim-v1/aim/v1/mlx/layers.py:25
↓ 1 callers
Class
SinCosPosEmbed
aim-v1/aim/v1/jax/layers.py:64
↓ 1 callers
Class
Transformer
aim-v1/aim/v1/torch/models.py:27
↓ 1 callers
Class
Transformer
aim-v1/aim/v1/mlx/models.py:14
↓ 1 callers
Class
Transformer
aim-v1/aim/v1/jax/models.py:14
Class
AIMForImageClassification
aim-v1/aim/v1/torch/models.py:105
Class
AIMMixin
aim-v1/aim/v1/mixins.py:11
Class
AIMv2TextMixin
aim-v2/aim/v2/mixins.py:28
Class
AIMv2VisionMixin
aim-v2/aim/v2/mixins.py:11
Class
Attention
aim-v1/aim/v1/torch/layers.py:177
Class
Attention
aim-v1/aim/v1/mlx/layers.py:151
Class
Attention
aim-v1/aim/v1/jax/layers.py:149
Class
AttentionPoolingClassifier
aim-v1/aim/v1/torch/layers.py:343
Class
AttentionPoolingClassifier
aim-v1/aim/v1/mlx/layers.py:276
Class
AttentionPoolingClassifier
aim-v1/aim/v1/jax/layers.py:254
Class
AverageLayers
aim-v1/aim/v1/torch/layers.py:324
Class
AverageLayers
aim-v1/aim/v1/mlx/layers.py:259
Class
AverageLayers
aim-v1/aim/v1/jax/layers.py:238
Class
Block
aim-v1/aim/v1/torch/layers.py:283
Class
Block
aim-v1/aim/v1/mlx/layers.py:223
Class
Block
aim-v1/aim/v1/jax/layers.py:208
Class
ExtractEOS
aim-v2/aim/v2/torch/layers.py:55
Class
ExtractEOS
aim-v2/aim/v2/mlx/layers.py:54
Class
ExtractEOS
aim-v2/aim/v2/jax/layers.py:59
Class
Identity
aim-v2/aim/v2/jax/layers.py:29
Class
LoraAttention
aim-v1/aim/v1/torch/layers.py:239
Class
MLP
aim-v1/aim/v1/torch/layers.py:256
Class
MLP
aim-v1/aim/v1/mlx/layers.py:196
Class
MLP
aim-v1/aim/v1/jax/layers.py:187
Class
MetricLogger
aim-v1/aim/v1/logger.py:92
Class
PatchEmbed
aim-v1/aim/v1/torch/layers.py:75
Class
PatchEmbed
aim-v1/aim/v1/mlx/layers.py:62
Class
PatchEmbed
aim-v1/aim/v1/jax/layers.py:25
Class
PrefixCausalAttention
aim-v1/aim/v1/torch/layers.py:217
Class
RMSNorm
aim-v2/aim/v2/jax/layers.py:68
Class
SwiGLUFFN
aim-v2/aim/v2/torch/layers.py:68
Class
SwiGLUFFN
aim-v2/aim/v2/mlx/layers.py:63
Class
SwiGLUFFN
aim-v2/aim/v2/jax/layers.py:83
Class
TestJAXBackend
aim-v2/tests/test_backend.py:69
Class
TestMLXBackend
aim-v2/tests/test_backend.py:28
Class
TextPreprocessor
aim-v2/aim/v2/torch/layers.py:30
Class
TextPreprocessor
aim-v2/aim/v2/mlx/layers.py:29
Class
TextPreprocessor
aim-v2/aim/v2/jax/layers.py:35
Class
ViTPreprocessor
aim-v1/aim/v1/torch/layers.py:110
Class
ViTPreprocessor
aim-v1/aim/v1/mlx/layers.py:98
Class
ViTPreprocessor
aim-v1/aim/v1/jax/layers.py:100