{"library":"flax","type":"library","category":null,"description":"Flax is a high-performance neural network library for JAX, designed for flexibility and ease of use. It provides building blocks for defining models, handling parameters, and managing training state within the JAX ecosystem. As of my last check, the current version is 0.12.6. Its release cadence is closely tied to JAX updates and major developments in the JAX ecosystem, with frequent minor and patch releases.","language":"python","status":"active","version":"0.12.6","tags":["jax","deep learning","neural networks","machine learning","google"],"last_verified":"Wed May 20","install":[{"cmd":"pip install flax jax[cpu]","imports":["import flax.linen as nn","from flax.core import FrozenDict","from flax.training import train_state"]},{"cmd":"pip install flax jax[cuda12_pip]","imports":[]}],"homepage":null,"github":"https://github.com/google/flax","docs":null,"changelog":null,"pypi":"https://pypi.org/project/flax/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":50,"avg_install_s":18.1,"avg_import_s":3.48,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/flax/compatibility"}}