metadata

title: README
emoji: ⚡
colorFrom: blue
colorTo: green
sdk: static
pinned: false

MLX Vision

A community org for model weights compatible with mlxim powered by MLX.

GitHub link: https://github.com/riccardomusmeci/mlx-image

These are weights converted from timm/torchvision and ready to be used.

How to install

pip install mlx-image

Models

To create a model with weights:

from mlxim.model import create_model

# loading weights from mlx-vision
model = create_model("resnet18")

# loading weights from local file
model = create_model("resnet18", weights="path/to/weights.npz")

More models will be uploaded aligned with MLX improvement by Apple team