Build a modular drop-in replacement library for PyTorch that implements tree-structured routing in place of traditional MLP blocks. This would allow researchers to easily swap standard layers for sparse, conditional computation versions in existing models.