diff --git a/.gitattributes b/.gitattributes
index 86abbea3cc238e08737df916d87cf09881f53252..819580c4587c224d6e9af409b1fb0eee7138d2a5 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -4470,3 +4470,19 @@ neuronxcc-2.19.8089.0+8ab9f450/MODULE_61e751d22792a685fb04+c2248236/model.neff f
 neuronxcc-2.19.8089.0+8ab9f450/MODULE_655a970d712c10190223+ca355898/model.neff filter=lfs diff=lfs merge=lfs -text
 neuronxcc-2.19.8089.0+8ab9f450/MODULE_655a970d712c10190223+ca355898/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
 neuronxcc-2.19.8089.0+8ab9f450/MODULE_7ab80dc216fdf22f33a4+431f5505/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.neff filter=lfs diff=lfs merge=lfs -text
+neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/wrapped_neff.hlo filter=lfs diff=lfs merge=lfs -text
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..0b2efd8b41c66b769b24aa55bb6aa78873872d18
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_966bdb17-7f93-4ceb-a1c7-4d2c18136f6b/compiler_workdir/GptOssRotaryEmbedding/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..f4e1b22333f914715f4de6a0204229b74f4978bc
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42c3e83c78fe60a7327b28a693dda20dc69469185f724e2c168bd11749f922eb
+size 2552
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..6edfc4e58310a8d827dd0d8673ef0eff5fee8d54
Binary files /dev/null and b/neuronxcc-2.20.9961.0+0acef03a/MODULE_1a16cd366a1d7ef23664+11fc5715/model.neff differ
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..af34d05001320150066426781c2da68f21c45664
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_4b6820a4-4f3d-4b43-b899-0517a4d75dea/compiler_workdir/GptOssAttention/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..5c4708c0e235abf5226da0520f4427874c15307b
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0012aa38c0ad5629d1dca88b509166a159aea72fa03f2c53a7397496e8caae2
+size 15486
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..ee9c2dbe3e872d42fe5985b27e2970fa3f264dd1
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0cf14b563a607d86db35cb720419c655411a77d997e793d814e9426e47a8557b
+size 7035904
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..05669b7763f7c327ce37e30bb6c8b78ed3db6784
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_2369e3a438054e465346+31987b5e/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71e3c78fbc1460820f6bad9684fb8d18f2b41f6a1c42d798f42ab5f24751607c
+size 7040678
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..d663103ee9f6939e4c15ea706aaa0749dab950de
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_820664a7-c308-4200-83c9-425cfb3ac2c3/compiler_workdir/GptOssAttention/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..5c4708c0e235abf5226da0520f4427874c15307b
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0012aa38c0ad5629d1dca88b509166a159aea72fa03f2c53a7397496e8caae2
+size 15486
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..bea3e5433a862fc76c5e748d1a60846696075f03
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:10b255e771e3307742ecfc41324606951efbee1a4bca9b323ad17f0ddff6e7e8
+size 7035904
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..4f824aec56c7aa0a3a279c52866eead7870c2336
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_4c4db5e0010f0372d97b+823a8c32/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77cceb09b188b769d43818810fdf2992fe2c8480c6127f60aafb5711c3cd618f
+size 7040678
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..868095cae7ae927117db25e12f01ad52f94a10e1
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_1826bd22-8f51-47aa-b13a-61d92beecc6e/compiler_workdir/GptOssRotaryEmbedding/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..f4e1b22333f914715f4de6a0204229b74f4978bc
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42c3e83c78fe60a7327b28a693dda20dc69469185f724e2c168bd11749f922eb
+size 2552
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..964b729ddcd598ea923d22d4cfde651f791f6ca6
Binary files /dev/null and b/neuronxcc-2.20.9961.0+0acef03a/MODULE_58d274cccb4429b9b598+bfd35efb/model.neff differ
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..c69fcd06668b1155067bead56512ce48a9e759ea
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_72562693-ef51-4b14-8822-715cf0a26a16/compiler_workdir/GptOssRotaryEmbedding/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..f4e1b22333f914715f4de6a0204229b74f4978bc
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42c3e83c78fe60a7327b28a693dda20dc69469185f724e2c168bd11749f922eb
+size 2552
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..809997ad3b0b7eac9ae459f5d1ca1cf553acf357
Binary files /dev/null and b/neuronxcc-2.20.9961.0+0acef03a/MODULE_73178bd87c5e177c20ef+a1041bc8/model.neff differ
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..edbd4dba34e05aef566d8daf77643684411e445e
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_64c2e137-316a-4b83-bf2e-40fe73a25a52/compiler_workdir/GptOssMLP/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..410d34061993036c6342de6e174dacbb5da5c11e
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d5f2d41ff3978b000dbaadc86b19e3b5d89c545c2e297c1c547898a3cecd7fe6
+size 7464
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..a936b6d29102cf53b4e887e7b04bbc1b2e356630
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d0613497ab4ae5ab6b64df96c931887a3da5f6b05f4bb9cbd12766c2bef671e
+size 175104
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..d3ec52ee57e820483e0823ed95dc1d0c20d8400c
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8e032341480934e99378+2cec3845/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eac16779439bd396701b79944cac0fbe6f5c6dd63834794bf343b826a4cdddd5
+size 178223
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..8b03102be91ec1d6b265febc1863ba29b266b6aa
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_a78f400e-1b0c-4da6-b8b1-46b4862bfea5/compiler_workdir/GptOssMLP/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..b1bf11eb353c97aea835abbdefbe872f163bb12d
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ed4162e12a10fa2681cbf4ade3964c4a8250863ae6a7114990e23a5e717808c
+size 7828
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..fb31ecba5318f37698c680dcb81c919481565648
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:38e2a4e9f6c2da972cf9113b6f01377380b7ff07ea54894a1a65ca228da449c4
+size 175104
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..09211d63739d7b609e6662e4e27695c0a13ff09d
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_8ff7f9d6056fe53873a7+1ab55550/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a6bd0eb1a7403135dfb76a78d383ef0daa16ce0f3dac3ac0d8e025cb209aa8e
+size 178235
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..43a9d4ba76530f019e8b56a32f48b0b2060badb2
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_1b93af70-d098-4bb8-a1b4-d86ddf4d919d/compiler_workdir/GptOssMLP/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..b1bf11eb353c97aea835abbdefbe872f163bb12d
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ed4162e12a10fa2681cbf4ade3964c4a8250863ae6a7114990e23a5e717808c
+size 7828
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..667b71a4b3393f5ef63efcedadd51a4430ff2ebd
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd9bd36287b8ac9eaeb3414723b83771115e001a73b859986115ecbe55b84260
+size 175104
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..574906f9b2e4e897735220ca4761c250a8b834ed
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_b4c2d00ed196a00612dc+e306bb4a/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a6424adba5729b095f20858022b617d24ade507dfddcada726ba443ee6dbb70
+size 178235
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..b4ccb1977057d45509a7b24f6a37f38fe0e3cec2
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_bcc1b010-1166-4ac1-9aed-b736cc9d9416/compiler_workdir/GptOssAttention/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..5c4708c0e235abf5226da0520f4427874c15307b
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0012aa38c0ad5629d1dca88b509166a159aea72fa03f2c53a7397496e8caae2
+size 15486
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..65b20a6fe6fa9e8b12e37f598ce845f1c6f9c9e3
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:73ca207befb039655ea3461e77141e6faef9da39c510d9461e49f3ffa59f238d
+size 7035904
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..4a06c7307d9bdacb789ee3efa6a9d311c59c3bc3
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ccfd54b298b70e659ef8+4e5f16bd/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:956078742e9c07e3901291eda331f7be6d5905eef063daa652f2ebe6e88333aa
+size 7040678
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..30a786d8ce59e3639facd0f9711736ef771a019f
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_2db7e0af-c125-42bb-966f-cdf85c6231ea/compiler_workdir/GptOssAttention/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..5c4708c0e235abf5226da0520f4427874c15307b
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0012aa38c0ad5629d1dca88b509166a159aea72fa03f2c53a7397496e8caae2
+size 15486
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..13202cc8aba7358a81d27fb83f004f39c513c2a5
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e644addf022b24b7aa3481363b80aca36faabae7f84a929916cb5c8fa657140f
+size 7035904
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..ab4531a3214b511ed94a64506d26d377360f3ad4
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ebb6b1c114879124d835+b2d01085/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a70eb424a5630eda05beef108102ceb6cfaf83b3ec6a7b11edb285d579acf42c
+size 7040678
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/compile_flags.json b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/compile_flags.json
new file mode 100644
index 0000000000000000000000000000000000000000..091fe1d73edf7943d8fa76ed5c57b485638f18d8
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/compile_flags.json
@@ -0,0 +1 @@
+["--target=trn1", "--enable-saturate-infinity", "--auto-cast=none", "--model-type=transformer", "-O1", "--logfile=/tmp/nxdi_test_cbd0f552-09a7-42b2-a294-8bcd6be298e1/compiler_workdir/GptOssMLP/_tp0_bk0/log-neuron-cc.txt", "--enable-internal-neff-wrapper"]
\ No newline at end of file
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.done b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.done
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.hlo_module.pb b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.hlo_module.pb
new file mode 100644
index 0000000000000000000000000000000000000000..b1bf11eb353c97aea835abbdefbe872f163bb12d
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.hlo_module.pb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ed4162e12a10fa2681cbf4ade3964c4a8250863ae6a7114990e23a5e717808c
+size 7828
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.neff b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.neff
new file mode 100644
index 0000000000000000000000000000000000000000..db803e988eab8a11b21e499f5f09ec2bafd86753
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/model.neff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f424e7c270e47ac6d0e686af3d49fc1988f48f8481614a640e1de7350081b34
+size 175104
diff --git a/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/wrapped_neff.hlo b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/wrapped_neff.hlo
new file mode 100644
index 0000000000000000000000000000000000000000..671b7392540f76518e0b3c49f78e924e7ecb023c
--- /dev/null
+++ b/neuronxcc-2.20.9961.0+0acef03a/MODULE_ec129ded949996c93bda+7447a6b0/wrapped_neff.hlo
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e34289e00282a2f766c0a4ed9586c294b0e62658a478ad72593e82d3c442d583
+size 178235