lr_mamba.jl update