# Core PyTorch with CUDA support (adjust CUDA version as needed)
torch>=2.0.0
torchvision>=0.15.0
torchaudio>=2.0.0

# Conditional Flow Matching library
torchcfm

# Neural ODE integration
torchdyn

# FID computation
cleanfid

# ODE solvers
torchdiffeq

# Command-line flags
absl-py

# Plotting and visualization
matplotlib
tensorboard

# Progress bars
tqdm

# Image processing
Pillow

# Data science utilities
numpy
scipy

# Optional: for enhanced logging
wandb
