Utils

• easydel.utils.__init__
  • DataClassArgumentParser
  • EasyQuantizer
  • GenerateRNG
  • JaxRNG
  • Timer
  • Timers
  • cache_compiles()
  • cjit()
  • compile_function()
  • get_logger()
  • load_compiled_fn()
  • save_compiled_fn()
• easydel.utils.analyze_memory
  • DeviceStats
  • MemoryMonitorClient
  • MemoryMonitorServer
  • SMPMemoryMonitor
  • start_client()
  • start_server()
• Checkpoint Managers
  • easydel.utils.checkpoint_managers.__init__
  • easydel.utils.checkpoint_managers.streamer
• easydel.utils.cli_helpers
  • Argu()
  • DataClassArgumentParser
  • make_choice_type_function()
  • string_to_bool()
• easydel.utils.compiling_utils
  • cache_compiles()
  • cjit()
  • compile_function()
  • get_hash_of_lowering()
  • get_safe_hash_int()
  • get_signature()
  • hash_fn()
  • is_jit_wrapped()
  • load_compiled_fn()
  • lower_function()
  • save_compiled_fn()
  • smart_compile()
• easydel.utils.graph_utils
  • get_module_from_path()
  • iter_module_search()
  • set_module_from_path()
• easydel.utils.helpers
  • ColorFormatter
  • DummyStream
  • LazyLogger
  • Timer
  • Timers
  • capture_time()
  • get_cache_dir()
  • get_logger()
  • quiet()
  • set_loggers_level()
• easydel.utils.parameters_transformation
  • convert_pytorch_tensor_to_jax()
  • float_tensor_to_dtype()
  • get_dtype()
  • get_torch()
  • jax2pt()
  • match_keywords()
  • module_to_huggingface_model()
  • module_to_torch()
  • process_tensor()
  • pt2jax()
  • torch_dict_to_easydel_params()
• easydel.utils.readme_generator
  • ModelInfo
  • ReadmeGenerator
• easydel.utils.rngs_utils
  • GenerateRNG
  • JaxRNG
  • next_rng()
• easydel.utils.traversals
  • MetaValueRecreator
  • StateValidationResult
  • create_graphdef()
  • deepcopy_model()
  • diffrentiate_state()
  • flatten_dict()
  • flatten_tree()
  • init_garphstate()
  • int_key_to_string()
  • is_flatten()
  • is_iterable()
  • merge_model_and_tree()
  • merge_state_and_tree()
  • named_tree_map()
  • nnx_init()
  • recreate_meta_values()
  • redefine_state()
  • refine_graphs()
  • specs_to_name_sharding()
  • string_key_to_int()
  • tree_apply()
  • tree_path_to_string()
  • unflatten_dict()
  • validate_state()