How to create this Bijector with tfb

Hello,
I am quite new in the field and in the use TF, so I probably ask a rather easy question, but I was wandering
how I can implement this transformation as a Bijector (all the “w” and “alpha” are >0)
$$T(z_i) = w_{i0} + \sum_{k=1}^K w_{ik}\rho(\alpha_{ik} z_i + \beta_{ik})$$
this is just to illustrate an how-to, for instance K=2 and z_i is either a 1-scalar feature or a 2-scalar features.