A vector search SQLite extension that runs anywhere!

Tools for merging pretrained large language models.

open source tools to generate mypy stubs from protobufs

A cross-platform browser ML framework.

All-in-one ml model exploration toolkit

This repository contains the official implementation of the research paper, "MobileCLIP: Fast Image-Text Models through Multi-Modal Reinforced Training" CVPR 2024

A package for running tflite files in MLX.

Graph Neural Network library made for Apple Silicon

C API for MLX

Implementation of RWKV using mlx

MLX support for the Open Neural Network Exchange (ONNX)

MLX: An array framework for Apple silicon

Load data directly into mlx

Ultra lightweight WebGL2 compute library

A marker based Augmented Reality library for Three.js

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

High-Resolution Image Synthesis with Latent Diffusion Models

firecracker-containerd enables containerd to manage containers as Firecracker microVMs

You like pytorch? You like micrograd? You love tinygrad! ❤️