Repository navigation

Revert "use Pytree as base type in nnx_wrappers" - #3

Merged

gulsumgudukbay merged 3 commits into

mainfrom

new_src_layout_modifications

Sep 23, 2025

Merged

Revert "use Pytree as base type in nnx_wrappers"#3
gulsumgudukbay merged 3 commits into
mainfrom
new_src_layout_modifications

Commits on Sep 23, 2025
- Revert "use Pytree as base type in nnx_wrappers"
  gulsumgudukbay
  committed
  a1b7b98
- Added a TODO to convert the import
  gulsumgudukbay
  authored
  70c715b
- Update nnx_wrappers.py
  gulsumgudukbay
  authored
  af61b46