Pedro Rodriguez c6ef4285e2 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:		2025-02-04 18:03:19 +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:03:19 +00:00