python313Packages.flax 0.12.10 single

Neural network library for JAX

Name: python3.13-flax-0.12.10
Package set: python313Packages
Homepage: https://github.com/google/flax
License: Apache-2.0
Source: pkgs/development/python-modules/flax/default.nix:130
Derivation: /nix/store/ays9hkl21c29ahy84361cqfma7cr2yck-python3.13-flax-0.12.10.drv
Repology: not reported as outdated · look up (as of 2026-10-07T10:23:38Z)

Maintainers 1

Handle	Name	Contact	Via
ndl	Alexander Tsvyashchenko	GitHub · ndl@endl.ch	listed on package

Dependencies 36

Package	As	Maintainers	Used by (transitive)	Status
python313 3.13.15	propagatedBuildInputs	mweinelt	13,660	team-only outdated
python313Packages.flaxlib 0.0.1	propagatedBuildInputs	GaetanLepage	26	single
python313Packages.jax 0.11.2	propagatedBuildInputs	ConnorBaker, ethancedwards8, GaetanLepage, samuela, SomeoneSerge, YorikSar	80	maintained
python313Packages.msgpack 1.2.1	propagatedBuildInputs	NickCao	1,647	single outdated
python313Packages.numpy 2.5.2	propagatedBuildInputs	doronbehar	5,137	single outdated
python313Packages.optax 0.2.8	propagatedBuildInputs	ndl	47	single
python313Packages.orbax-checkpoint 0.12.6	propagatedBuildInputs	fabaff	39	single
python313Packages.pyyaml 6.0.3	propagatedBuildInputs	dotlambda	5,105	single
python313Packages.rich 15.0.0	propagatedBuildInputs	risicle	3,870	single
python313Packages.tensorstore 0.1.84	propagatedBuildInputs	samuela	42	single outdated
python313Packages.treescope 0.1.10	propagatedBuildInputs	GaetanLepage	26	single
python313Packages.typing-extensions 4.16.0	propagatedBuildInputs	pmiddend	7,553	single
ensureNewerSourcesForZipFilesHook 26.05pre-git	nativeBuildInputs		74,596	unmaintained setup hook
python313Packages.cloudpickle 3.1.2	nativeBuildInputs		3,929	unmaintained
python313Packages.einops 0.8.2	nativeBuildInputs	yl3dy	89	single
python313Packages.keras 3.15.1	nativeBuildInputs	GaetanLepage	37	single
python313Packages.pypaBuildHook 26.05pre-git	nativeBuildInputs		11,598	unmaintained setup hook
python313Packages.pypaInstallHook 26.05pre-git	nativeBuildInputs		11,625	unmaintained setup hook
python313Packages.pytest-xdist 3.8.0	nativeBuildInputs	dotlambda	7,922	single
python313Packages.pytestCheckHook 26.05pre-git	nativeBuildInputs		10,182	unmaintained setup hook
python313Packages.pythonCatchConflictsHook 26.05pre-git	nativeBuildInputs		11,609	unmaintained setup hook
python313Packages.pythonImportsCheckHook 26.05pre-git	nativeBuildInputs		11,642	unmaintained setup hook
python313Packages.pythonMetadataCheckHook 26.05pre-git	nativeBuildInputs	dotlambda	11,597	single setup hook
python313Packages.pythonNamespacesHook 26.05pre-git	nativeBuildInputs		11,634	unmaintained setup hook
python313Packages.pythonOutputDistHook 26.05pre-git	nativeBuildInputs		74,564	unmaintained setup hook
python313Packages.pythonRemoveBinBytecodeHook 26.05pre-git	nativeBuildInputs		74,584	unmaintained setup hook
python313Packages.pythonRemoveTestsDirHook 26.05pre-git	nativeBuildInputs		11,634	unmaintained setup hook
python313Packages.pythonRuntimeDepsCheckHook 26.05pre-git	nativeBuildInputs		11,611	unmaintained setup hook
python313Packages.setuptools 83.0.0	nativeBuildInputs	mweinelt	11,500	team-only outdated
python313Packages.setuptools-scm 10.0.5	nativeBuildInputs	NickCao	10,397	single outdated
python313Packages.sphinx 9.1.0	nativeBuildInputs		6,025	unmaintained
python313Packages.tensorflow 2.21.0	nativeBuildInputs		61	unmaintained
python313Packages.torch 2.13.0	nativeBuildInputs	caniko, GaetanLepage, LunNova, teh, thoughtpolice, tscholak	399	maintained outdated
python313Packages.wrapPython 26.05pre-git	nativeBuildInputs		11,634	unmaintained setup hook
bashNonInteractive 5.3p15	other input	andir, balsoft, infinisil, pyrox0	102,441	maintained outdated
stdenv 26.05pre-git	other input		90,330	team-only

Used by 9 directly, 25 transitively

Package	As	Maintainers	Used by (transitive)	Status
python313Packages.gymnasium 1.3.0	nativeBuildInputs	GaetanLepage	10	single outdated
python313Packages.numpyro 0.22.0	nativeBuildInputs	fabaff	7	single
python313Packages.brax 0.14.2	propagatedBuildInputs	nim65s	1	single
python313Packages.clu 0.0.12	propagatedBuildInputs	GaetanLepage	0	single
python313Packages.evosax 0.3.1	propagatedBuildInputs	GaetanLepage	0	single
python313Packages.skrl 2.1.0	nativeBuildInputs	bcdarwin	0	single
python313Packages.transformer-engine 2.19	propagatedBuildInputs	ConnorBaker, ethancedwards8, GaetanLepage, samuela, SomeoneSerge, YorikSar	0	maintained broken outdated
python313Packages.transformer-engine-jax 2.19	propagatedBuildInputs	ConnorBaker, ethancedwards8, GaetanLepage, samuela, SomeoneSerge, YorikSar	0	maintained broken outdated
python313Packages.waymax 0-unstable-2025-05-30	propagatedBuildInputs	samuela	0	single unfree