google / google/torchax

How to integrate Torch model with Flax nnx?

Open
#94 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
243
Forks
38
Avg merge
15h 32m
Merged PRs (30d)
2

Description

Hi,

Wanted to use one of the Amazon Chronos models as an input to my FlaxNNX model.
I was curious if there were any suggestions / examples on how I should go about it, and also allow updating of the Chronos models weights?

Thanks,
Jorge

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.