/mimm

MLX Image Models

Primary LanguagePython

MLX Image Models

Introduction

This repository contains implementations for image models in MLX with functions to load pretrained weights from pytorch. Example training code is provided in mimm/scripts.

Models

Classification

Acknowledgements

This repository is based on the following repositories: