Skip to content

implement pytorch shim with benchmark tests#32

Draft
gnieuwenhuis wants to merge 1 commit into
large-memory-fixfrom
pytorch
Draft

implement pytorch shim with benchmark tests#32
gnieuwenhuis wants to merge 1 commit into
large-memory-fixfrom
pytorch

Conversation

@gnieuwenhuis
Copy link
Copy Markdown
Contributor

In order to enable deeper experiments, creating a simple shim that defined a pytorch model instead of an sklearn MLP architecture.

The goal here is to create a nearly identical model using pytorch, including unit tests to ensure similarity between the two models.

Signed-off-by: Greg Nieuwenhuis <26285069+gnieuwenhuis@users.noreply.github.com>
@gnieuwenhuis gnieuwenhuis linked an issue May 25, 2026 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Change underlying model infrastructure from Sklearn to Pytorch

1 participant