var('s','t','T')
A=matrix(A_elements_dict);

f=A(s=t/T)

reset('i');
reset('x');

def Sch(t,y):
    return [ -i*sum(f[j,k]*y[j] for k in range(2^L)) for j in range(2^L)]

NL1=[1 for i in range(2^L)];

TS=ode_solver()
TS.algorithm = "rk4"
TS.function=Sch
TS.y_0 = NL1
TS.ode_solve(t_span=[0,T], num_points=100)
print TS.solution;