# TensorFlow Brain-to-Text Requirements for TPU v5e-8
# Install with: pip install -r requirements_tf.txt

# Core TensorFlow and TPU support
tensorflow>=2.15.0
tensorflow-text>=2.15.0

# Data processing
numpy>=1.21.0
h5py>=3.7.0
scipy>=1.9.0

# Configuration and utilities
omegaconf>=2.3.0
pyyaml>=6.0

# Logging and monitoring
tensorboard>=2.12.0
wandb>=0.15.0  # Optional: for experiment tracking

# Audio processing (for phoneme analysis)
librosa>=0.10.0

# Development and testing
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
flake8>=5.0.0

# Optional: Jupyter for analysis notebooks
jupyter>=1.0.0
matplotlib>=3.5.0
seaborn>=0.11.0
pandas>=1.4.0

# Memory optimization
psutil>=5.8.0

# Note: For TPU v5e-8 environments, TensorFlow should be pre-installed
# with TPU support. This requirements file is for additional dependencies.