flax
Flax: A neural network library for JAX designed for flexibility
- Category
- Python packages
pip install flax Official docs outline
flax.readthedocs.ioExtra Credit builds an outline of flax's docs the first time your work uses it. Until then, start at the official documentation.
More in Python packages
All 9,553-
typed-settings Typed settings based on attrs classes
-
django-crispy-forms Best way to have Django DRY forms
-
pycron Simple cron-like parser, which determines if current datetime matches conditions.
-
django-reversion An extension to the Django web framework that provides version control for model instances.
-
pytest-profiling Profiling plugin for py.test
-
inline-snapshot golden master/snapshot/approval testing library which puts the values right into your source code
-
comfyui-frontend-package
-
grandalf Graph and drawing algorithms framework
-
pytest-docker Simple pytest fixtures for Docker and Docker Compose based tests
-
mcp-server-qdrant MCP server for retrieving context from a Qdrant vector database
-
mdutils Useful package for creating Markdown files while executing python code.
-
itemadapter Common interface for data container classes