In the second error you are probably doing something like this list(5) instead of doing list[5], or some other variation inside the indexing of the list that its used for tuples.
Try to read a bit about lists and tuples in python.
OK. I have fixed partial errors on C4 and C5, the problem in C4 is easily solved using
`jnp.shape(?. , (-1, ?,?,?)`
For C_5, I believe there is a small mistake in the original template because it is
type or paste code h tl.Branch( # creates three towers for one input, takes activations and creates queries keys and values here
[?, ?] # queries
[?, ?] # keys
[?, ?] # values
)
but it should be
type or paste code here
tl.Branch( # creates three towers for one input, takes activations and creates queries keys and values
[?, ?], # queries
[?, ?], # keys
[?, ?], # values
)