iliailmer / tinyflow

Flow Matching implemented in tinygrad

Flow Matching using tinygrad

• Flow Matching using tinygrad
  • Introduction
  • 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 to run this project.

uv run example.py

To-Do

Time permitting, I plan to add the following implementations:

• more path objects

• non-euclidean flow matching

• discrete flow matching

References

• Flow Matching for Generative Modeling
• Flow Matching Guide and Code