[Backport r0.8] Fix LAMB optimizer regex parsing

[bot-of-gabrieldemarmiesse]

Backport ce16e62 from #1532

[googlebot]

We found a Contributor License Agreement for you (the sender of this pull request), but were unable to find agreements for all the commit author(s) or Co-authors. If you authored these, maybe you used a different email address in the git commits than was used to sign the CLA (login here to double check)? If these were authored by someone else, then they will need to sign a CLA as well, and confirm that they're okay with these being contributed to Google.
In order to pass this check, please resolve this problem and then comment @googlebot I fixed it.. If the bot doesn't comment, it means it doesn't think anything has changed.

ℹ️ Googlers: Go here for more info.

[seanpmorgan]

LGTM

[jarednielsen]

Thanks! Is there an ETA for publishing 0.8.5 to PyPi?

[seanpmorgan]

Thanks! Is there an ETA for publishing 0.8.5 to PyPi?

We're running into VM disk space limits on the GitHub action. Currently discussing the best way forward. May require a 0.9 release with new features + bugfixes

Will update an ETA shortly

[jarednielsen]

Good to know. In the meantime, would using the latest tfa-nightly package work?

[seanpmorgan]

Good to know. In the meantime, would using the latest tfa-nightly package work?

Yes it should. Will update here with an ETA for 0.9 release after our monthly meeting today

[seanpmorgan]

@jarednielsen 0.9 has been published.

[jarednielsen]

Thank you!