Skip to content

Instantly share code, notes, and snippets.

View shivammehta25's full-sized avatar
:octocat:
this animal in his natural habitat

Shivam Mehta shivammehta25

:octocat:
this animal in his natural habitat
View GitHub Profile
@francois-rozet
francois-rozet / flow_matching.py
Last active May 8, 2024 04:57
Flow Matching in 100 LOC
#!/usr/bin/env python
import math
import matplotlib.pyplot as plt
import torch
import torch.nn as nn
from sklearn.datasets import make_moons
from torch import Tensor
from tqdm import tqdm