40 lines
834 B
Plaintext
40 lines
834 B
Plaintext
# 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. |