https://github.com/iliailmer/tinyflow

Flow Matching implemented in tinygrad
https://github.com/iliailmer/tinyflow

flowmatching tinygrad

Last synced: 2 months ago
JSON representation

Flow Matching implemented in tinygrad

• Host: GitHub
• URL: https://github.com/iliailmer/tinyflow
• Owner: iliailmer
• License: mit
• Created: 2025-01-20T19:22:09.000Z (11 months ago)
• Default Branch: main
• Last Pushed: 2025-08-13T00:14:25.000Z (5 months ago)
• Last Synced: 2025-10-13T14:36:37.842Z (2 months ago)
• Topics: flowmatching, tinygrad
• Language: Python
• Homepage:
• Size: 325 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# Flow Matching using `tinygrad`

[![Tests](https://github.com/iliailmer/tinyflow/actions/workflows/tests.yml/badge.svg)](https://github.com/iliailmer/tinyflow/actions/workflows/tests.yml)

[![Lint](https://github.com/iliailmer/tinyflow/actions/workflows/lint.yml/badge.svg)](https://github.com/iliailmer/flow_matching_tinygrad/actions/workflows/lint.yml)

[![Python 3.11+](https://img.shields.io/badge/python-3.11+-blue.svg)](https://www.python.org/downloads/)

[![License](https://img.shields.io/badge/license-MIT-green.svg)](LICENSE)

- [Flow Matching using `tinygrad`](#flow-matching-using-tinygrad)

  - [Introduction](#introduction)

  - [References](#references)

  

## Introduction

This project is my exploration into flow matching

algorithms using `tinygrad` library instead of traditional

`pytorch` implementations.

This is a learning experience for

understanding both the flow matching and the library.

## Running the Code

I highly recommend using [`uv` tool](https://github.com/astral-sh/uv) to run this project.

```python

uv run example.py

```

## To-Do

Time permitting, I plan to add the following implementations:

- [x] more path objects

- [ ] non-euclidean flow matching

- [ ] discrete flow matching

## References

- [Flow Matching for Generative Modeling](https://arxiv.org/pdf/2210.02747)

- [Flow Matching Guide and Code](https://arxiv.org/pdf/2412.06264)