flax
[nnx] default runtime attributes + train/eval methods
#3875

Merged

[nnx] default runtime attributes + train/eval methods #3875

cgarciae merged 1 commit into nnx-v0.1 from nnx-runtime-defaults

[nnx] default runtime attributes + train/eval methods

1dea4990

cgarciae force pushed to 1dea4990 1 year ago

cgarciae changed the base branch from nnx-update-transforms to nnx-v0.1 1 year ago

cgarciae changed the base branch from nnx-v0.1 to main 1 year ago

cgarciae changed the base branch from main to nnx-v0.1 1 year ago

cgarciae merged 1dea4990 into nnx-v0.1 1 year ago

cgarciae deleted the nnx-runtime-defaults branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone