https://github.com/sauravmaheshkar/mlx-ssl
self-supervised learning on apple silicon
https://github.com/sauravmaheshkar/mlx-ssl
data2vec mlx self-supervised-learning
Last synced: 4 months ago
JSON representation
self-supervised learning on apple silicon
- Host: GitHub
- URL: https://github.com/sauravmaheshkar/mlx-ssl
- Owner: SauravMaheshkar
- License: mit
- Created: 2025-02-06T18:15:15.000Z (8 months ago)
- Default Branch: main
- Last Pushed: 2025-05-18T00:27:59.000Z (5 months ago)
- Last Synced: 2025-06-14T20:58:47.151Z (4 months ago)
- Topics: data2vec, mlx, self-supervised-learning
- Language: Python
- Homepage:
- Size: 65.4 KB
- Stars: 10
- Watchers: 1
- Forks: 0
- Open Issues: 4
-
Metadata Files:
- Readme: .github/README.md
- Contributing: .github/CONTRIBUTING.md
- Funding: .github/FUNDING.yml
- License: LICENSE
- Code of conduct: .github/CODE_OF_CONDUCT.md
- Codeowners: .github/CODEOWNERS
Awesome Lists containing this project
README
## MLX SSL
Python package for self-supervised learning on Apple Silicon using MLX
### Model Implementations
1. [SimCLR](https://arxiv.org/abs/2002.05709)
```python
from mlx_ssl.models import SimCLR
# Available Models:
# * mlx-community/simclrv1-imagenet1k-resnet50-1x
# * mlx-community/simclrv1-imagenet1k-resnet50-2x
# * mlx-community/simclrv1-imagenet1k-resnet50-4x
model = SimCLR.from_pretrained(
"mlx-community/simclrv1-imagenet1k-resnet50-1x"
)
```
2. [Data2Vec](https://arxiv.org/abs/2202.03555)
```python
from mlx_ssl.models import Data2VecVisionForImageClassification
model = Data2VecVisionForImageClassification.from_pretrained(
"mlx-community/data2vec-vision-large-ft1k"
)
```