-
Shiva Shahrokhi authored
flax version needs to be updated in requirements, I got this error while trying: "TypeError: __init__() got an unexpected keyword argument 'step_offset'" , and that was because of the flax version mismatch PiperOrigin-RevId: 356817800
fce923e9
This project manages its dependencies using pip.
Learn more