piwheels

flaxmodels

A collection of pretrained models in Flax.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install flaxmodels

Dependencies

• jaxlib
• filelock
• flax
• regex
• tqdm
• pillow
• numpy
• h5py
• dataclasses
• packaging
• jax
• requests

Releases

Version	Released	Bookworm Python 3.11	Trixie Python 3.13	Files
0.1.3	2023-08-11
flaxmodels-0.1.3-py3-none-any.whl (113 KB)
0.1.2	2022-07-14
flaxmodels-0.1.2-py3-none-any.whl (113 KB)
0.1.1	2021-09-10
flaxmodels-0.1.1-py3-none-any.whl (99 KB)
0.1.0	2021-05-23
flaxmodels-0.1.0-py3-none-any.whl (96 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue