This commit is contained in:
Zchen
2025-10-15 16:55:52 +08:00
parent b466e97463
commit 7965f7dbfe
16 changed files with 3571 additions and 954 deletions

View File

@@ -0,0 +1,40 @@
# 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.