Help with the GA and GY Networks

[angeruzzi]

Hi Valentyn,

First of all, I would like to congratulate you and the other authors on the excellent work presented in the paper "Causal Transformer for Estimating Counterfactual Outcomes".

I am amazed by the quality and depth of the research, and especially grateful for making the source code available in such a structured and didactic manner.

I am studying the code with great interest as I intend to use your work as an inspiration for my own research. However, I am having difficulty locating the implementation of the GA and GY networks mentioned in the paper. Could you please help me identify where these networks are implemented in the source?

Thank you in advance for your attention and for the significant contribution to the research community.

Best regards, Alessandro Angeruzzi