Utilities

This section contains documentation for utilities.

Modules

Code Analysis

• dependency_analyzer

Image

• color
• metrics
• transforms

Io

• file
• formats
• serialization

Jax

• device
• dtype
• flax_utils
• prng
• shapes

Logging

• logger
• metrics
• mlflow
• wandb
• file_utils

Numerical

• math
• stability
• stats

Profiling

• memory
• performance
• xprof

Text

• metrics
• postprocessing
• processing

Utils

• env
• registry
• timer
• types

Visualization

• attention_vis
• image_grid
• latent_space
• plotting
• protein

Statistics

• Total modules: 38
• Total classes: 9
• Total functions: 79