Tianyuan Zhang1,
Sai Bi2,
Yicong Hong2,
Kai Zhang2,
Fujun Luan2,
Songlin Yang1,
Kalyan Sunkavalli2,
William T. Freeman1,
Hao Tan2
1MIT 2Adobe Research
We provide minimal implementations for a LaCT layer in minimal_implementations/. This implementation serves as a starting point for understanding, modifying, and creating your own version of LaCT.
If you find this codebase useful for your research, please kindly cite our paper:
@article{zhang2025test,
title={Test-time training done right},
author={Zhang, Tianyuan and Bi, Sai and Hong, Yicong and Zhang, Kai and Luan, Fujun and Yang, Songlin and Sunkavalli, Kalyan and Freeman, William T and Tan, Hao},
journal={arXiv preprint arXiv:2505.23884},
year={2025}
}
Python
99.6%
Tianyuan Zhang1,
Sai Bi2,
Yicong Hong2,
Kai Zhang2,
Fujun Luan2,
Songlin Yang1,
Kalyan Sunkavalli2,
William T. Freeman1,
Hao Tan2
1MIT 2Adobe Research
We provide minimal implementations for a LaCT layer in minimal_implementations/. This implementation serves as a starting point for understanding, modifying, and creating your own version of LaCT.
If you find this codebase useful for your research, please kindly cite our paper:
@article{zhang2025test,
title={Test-time training done right},
author={Zhang, Tianyuan and Bi, Sai and Hong, Yicong and Zhang, Kai and Luan, Fujun and Yang, Songlin and Sunkavalli, Kalyan and Freeman, William T and Tan, Hao},
journal={arXiv preprint arXiv:2505.23884},
year={2025}
}
Python
99.6%