Pedro Rodriguez bc39591032 Several changes to enable entropy model training/eval ... Summary: - Make arrow iterator able to read from jsonl files, the entropies are omitted in this case - Make the data/checkpoint code fsspec compatible - Fix issues with all reduce with non-bf16 in dist_sum and norm computation. - Minimal fixes to get eval to run, it is slow currently - Add bpb numbers during training Test Plan: Run ``` torchrun --nproc-per-node 8 -m bytelatent.train config=internal/configs/entropy_model.yaml eval=null max_steps=10100 ``` ``` python -m bytelatent.train config=internal/configs/s3_debug.yaml eval=null ``` ``` torchrun --nproc-per-node 8 -m bytelatent.train config=internal/configs/s3_debug.yaml eval=null ```		2025-02-04 18:19:49 +00:00
..
__init__.py	Initial commit	2024-12-12 15:32:30 -08:00
data_pipeline.py	Initial commit	2024-12-12 15:32:30 -08:00
fsspec_target.py	Changes for training entropy model and correcting attention in local models (#25)	2025-01-17 14:23:01 -08:00
parallel_entropies.py	Initial commit	2024-12-12 15:32:30 -08:00
preprocess_entropies.py	Several changes to enable entropy model training/eval	2025-02-04 18:19:49 +00:00