# requirements.txt

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

# Data processing and serialization
cloudpickle
pillow

# Configuration and project management
omegaconf
gitpython

# Monitoring and logging
tensorboard
psutil

# Networking
requests
