Skip to content

v0.2.0

@armbrusl armbrusl tagged this 17 Apr 15:26
- Remove FlaxModelWrapper, FlaxNNXWrapper from common.py
- Remove Flax NNX/Linen paths from trace_compiler.py
- Remove Flax LoRA wrappers from lora_linear.py
- Add per-target LoRA specs with regex matching
- Add per-model policy transparency report in core.py
- nn.wrap() now requires eqx.Module (TypeError on Flax)
- Pin foundax>=0.1.4 (returns eqx.Module directly)
- Delete test_flax_wrapper.py, remove NNX integration tests
Assets 2
Loading