PyRank

Flax Nnx Python Packages

Python packages with the GitHub topic flax-nnx. Sorted by relevance, with stars and monthly downloads.
avitai
datarax

A Differentiable Data Pipeline Framework for JAX

5K 1 0
avitai
calibrax

Unified benchmarking and profiling framework for the JAX scientific ML ecosystem. Timing, GPU/energy monitoring, FLOPS counting, roofline analysis, statistical testing, regression detection, and CI integration.

4K 1 0
avitai
avitai-artifex

A research-focused modular generative modeling library built on JAX/Flax NNX

3K 1 0
avitai
diffbio

End-to-end differentiable bioinformatics pipelines built on JAX/Flax NNX

1K 1 0
Xrenya
jaxnn

JaxNN: Foundation Models in JAX/Flax

520 0 0
tillahoffmann
flax-metrics

Flax NXX implementation of common metrics.

426 1 0