Skip to content

Streamline parameter group implementation#13

Open
nandantumu wants to merge 1 commit into
mainfrom
codex/convert-library-to-support-jax
Open

Streamline parameter group implementation#13
nandantumu wants to merge 1 commit into
mainfrom
codex/convert-library-to-support-jax

Conversation

@nandantumu

Copy link
Copy Markdown
Owner

Summary

  • collapse the parameter utilities into a minimal base class and sample container without training state
  • simplify PointParameterGroup to deterministic dictionary-backed values that share sampling and evaluation
  • remove unused distribution and bounded parameter group implementations and only export the slimmed API

Testing

  • pytest

https://chatgpt.com/codex/tasks/task_e_68e01771a34c8320a8517ad6c0718279

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant