Adrien Toupet b40f26167c Fix MPS compatibility: disable antialias for MPS tensors, fix bfloat16 arange (#354)

2025-12-03 13:09:59 -05:00

..

diffusion

Fix MPS compatibility: disable antialias for MPS tensors, fix bfloat16 arange (#354)

2025-12-03 13:09:59 -05:00

distributed

fix: use canonical torch.backends.mps.is_available() for reliable MPS detection

2025-11-28 11:49:43 -05:00

__init__.py

fix import

2025-07-01 13:33:50 +02:00

cache.py

fix import

2025-07-01 13:33:50 +02:00

config.py

refactor: Add indent_level parameter to debug logging system

2025-10-19 10:13:17 -04:00

decorators.py

fix: make FSDP imports conditional for AMD ROCm compatibility

2025-10-26 23:49:52 -04:00

half_precision_fixes.py

refactor: 42, centralize float32 precision utilities and improve logging

2025-10-16 14:30:47 -04:00

logger.py

fix import

2025-07-01 13:33:50 +02:00

partition.py

fix import

2025-07-01 13:33:50 +02:00

seed.py

fix import

2025-07-01 13:33:50 +02:00