flax

3824622c - Merge pull request #3590 from jianyizh:jianyi/embed