cgarciae/nnx

Test sharing lifted Modules

cgarciae opened this issue · 0 comments

# foo.bar is transformed by vmap
shared_thing = foo.bar.shared_thing
# baz.qup is transformed by vmap
baz.qup.shared_thing = shared_thing