CarolinePascal ca6d764107 fix(autocast): gate autocast on AMP-capable devices ...

Add get_safe_autocast_context helper that only enters torch.autocast on
devices supporting AMP (cuda, xpu, cpu) and falls back to a no-op on mps
and unknown backends. Route the previously unconditional/underspecified
autocasts (vla_jepa, groot, molmoact2, lerobot_eval) through it so
autocast can be requested unconditionally without breaking on unsupported
devices.

2026-07-03 11:22:33 +02:00

..

test_collate.py

Add extensive language support (#3467)

2026-05-19 14:46:11 +02:00

test_device_utils.py

fix(autocast): gate autocast on AMP-capable devices

2026-07-03 11:22:33 +02:00

test_encoding_utils.py

chore(utils): move encoding utils and process to their respective modules (#2029)

2025-09-24 16:47:37 +02:00

test_foxglove_visualization.py

feat(visualization): add foxglove support (#3902)

2026-07-01 18:39:32 +02:00

test_hub.py

feat(train): run training remotely on HF Jobs via --job.target (#3856)

2026-06-29 17:59:33 +02:00

test_io_utils.py

chore(docs): add missing license headers (#2140)

2025-10-08 14:27:52 +02:00

test_keyboard_input.py

feat(utils): display-independent keyboard controls for recording (Wayland / headless / macOS) (#3875)

2026-06-25 10:58:39 +02:00

test_logging_utils.py

feat(training): bump accelerate + use reduction types for tracked metrics in a multi rank setup (#3773)

2026-06-11 19:07:28 +02:00

test_process.py

RL stack refactoring (#3075)

2026-05-12 15:49:54 +02:00

test_random_utils.py

chore(docs): add missing license headers (#2140)

2025-10-08 14:27:52 +02:00

test_replay_buffer.py

RL stack refactoring (#3075)

2026-05-12 15:49:54 +02:00

test_rerun_visualization.py

fix(depth unit): adding input depth unit storage in the dataset metadata (#3899)

2026-07-02 11:53:13 +02:00

test_sample_weighting.py

Reward models refactor (#3142)

2026-04-28 17:56:24 +02:00

test_train_utils.py

feat(train): run training remotely on HF Jobs via --job.target (#3856)

2026-06-29 17:59:33 +02:00

test_visualization_utils.py

feat(visualization): add foxglove support (#3902)

2026-07-01 18:39:32 +02:00