Irakli Salia

Tensors and Dynamic neural networks in Python with strong GPU acceleration

GEMMs with metal

A Python library that replaces PyTorch's matrix multiplication on Apple Silicon Macs with custom Metal GPU code, achieving 2-3x faster performance for 32-bit floats and matching PyTorch for 16-bit formats used in large language models.

Torch extension for calculating the Intersection over Union (IoU) for quadrilaterals, supports CPU and CUDA tensors. Handles both 1-to-1 and M-to-N matches, returning an IoU matrix with M rows and N columns.

Repository for finetuning DETR on SKU110K dataset with num_queries > 100

Repository for implementing computer vision papers from scratch

Repository for answers for exercises in Programming Massively Parallel Processors book

Repository for Sparse MPS matrices paper

Implementation of vision transformer with register. Replication of the following paper from scratch: https://arxiv.org/pdf/2309.16588

A high-throughput and memory-efficient inference and serving engine for LLMs

MLX: An array framework for Apple silicon

🚧 Accepting Task Submissions 🚧

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

No description.

Simple, safe way to store and distribute tensors

Perf monitoring CLI tool for Apple Silicon

Datasets, Transforms and Models specific to Computer Vision

GAIA olympiad

Machine learning metrics for distributed, scalable PyTorch applications.

Code for lectures for GAIA

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

No description.

Geometric Computer Vision Library for Spatial AI

Serve, optimize and scale PyTorch models in production

Repository for implementing bloom filters in C

Project for distributed systems

Mask2Former for Parsing (M2FP)

Repo for public images

Background Remover lets you Remove Background from images and video using AI with a simple command line interface that is free and open source.

Repository for a master's course - Advanced Algorithms and Data Structures

Repository for pracitcing leetcode questions

Repository for SQL Lecture files

Library for ML Scripts