flax

4d075e53 - Copybara import of the project:

Copybara import of the project: -- c9e63a7b0ae87fd61c0b10030ba33bd9a7c7a92d by Cristian Garcia <cgarcia.e88@gmail.com>: add compact_name_scope PiperOrigin-RevId: 600373748