DGX_AI / trainer

Commit History

fix: patch torch._pytree.register_constant + tighten torchao/transformers pins
a38ca5e

vasiuuu commited on

Upgrade to PyTorch 2.6.0 base image for native torchao support
ab48b50

vasiuuu commited on

Patch all missing torch int1-int7 dtypes for torchao compat
6e53e4a

vasiuuu commited on

Add explicit torch.int1 workaround
2ca858a

vasiuuu commited on

Fix reward_fn sandbox signature
6718f28

vasiuuu commited on

Enable WandB logging and add API key
dbf2d43

vasiuuu commited on

Initial commit for CodeForge GRPO training
acf77ab

vasiuuu commited on