# requirements.txt

# Core ML/JAX dependencies
attrs
chex
dpshdl
equinox
importlib-resources
jax
jaxtyping
optax
orbax-checkpoint

# Data processing and serialization
pillow

# Configuration and project management
omegaconf
gitpython

# Monitoring and logging
tensorboard
psutil

# Networking
requests
