Thanks for the advice @alvarobartt - I think this last commit addresses the things you mentioned

Great! Thanks for your effort here @ihh, just mentioned a couple things I'd like to change, and also to let you know that you are using a different user to push the commits (@pixelzoo) in case you want to change that and force-push with a different user

I think the committing user was showing as @pixelzoo because my email address was associated with that account. Removing that association and associating that email address with @ihh instead seems to have fixed the problem (unless I am looking at the wrong thing)

It seems that the CI's having a hard time finding the jax and jaxlib latest version i.e. 0.4.14, I'll try to have a look into it later today! Thanks again for your contribution @ihh

hi, could you possibly merge this or advise on how to fix the PR? This seems to be the easiest way to use safetensors with jax that I can find but it's currently uninstallable with pip because jaxlib 0.3.25 is no longer available

EDIT: I missed the comment directly above this. sadly I don't know how to solve this so not sure I can unblock this PR myself.

EDIT: I missed the comment directly above this. sadly I don't know how to solve this so not sure I can unblock this PR myself.

Oh hi here @perrybleiberg I'm sorry to hear that, sorry I left this aside, I'll try to fix this over the weekend! I'm glad that you're using / trying to use safejax 🤗

thank you so much!