Yufan Zhuang

MetaTree. Official implementation of MetaTree: Learning a Decision Tree Algorithm with Transformers

test_time_recursive_thinking. Python

knowledge_flow. Official Implementation of Knowledge Flow Prompting

mixinputs. Official implementation for Text Generation Beyond Discrete Token Sampling

vector-icl. Official implementation of Vector-ICL: In-context Learning with Continuous Vector Representations (ICLR 2025)

MRI-Reconstruction-with-Sparse-Optimization. Magnetic resonance imaging (MRI) images are known to be sparse. This is an implementation using non-convex penalty function that encourages sparsity.

AgenticLU. Official implementation of Self-Taught Agentic Long Context Understanding (ACL 2025).

wavspa. WavSpA: Wavelet Space Attention for Enhancing Transformer's Long Sequence Learning

rocm_tips. Tips for building and using DL packages for AMD ROCM