CHANGELOG.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
PYPI_README.md
README.md
pyproject.toml
src/nnx/__init__.py
src/nnx/_artifacts.py
src/nnx/_builders.py
src/nnx/_config.py
src/nnx/_confusion.py
src/nnx/_metrics.py
src/nnx/_optional.py
src/nnx/_probability.py
src/nnx/_scheduler_clock.py
src/nnx/_step_helpers.py
src/nnx/_update_engine.py
src/nnx/_validation.py
src/nnx/abstention.py
src/nnx/benchmarking.py
src/nnx/bundles.py
src/nnx/calibration.py
src/nnx/comparison.py
src/nnx/compilation.py
src/nnx/components.py
src/nnx/data_splits.py
src/nnx/distributed.py
src/nnx/export_conformance.py
src/nnx/graph_tasks.py
src/nnx/history.py
src/nnx/link_tasks.py
src/nnx/lm_tasks.py
src/nnx/lr_finder.py
src/nnx/models.py
src/nnx/monitors.py
src/nnx/objectives.py
src/nnx/optimizers.py
src/nnx/plans.py
src/nnx/precision.py
src/nnx/prediction.py
src/nnx/preprocessing.py
src/nnx/provenance.py
src/nnx/py.typed
src/nnx/ranking.py
src/nnx/result.py
src/nnx/search.py
src/nnx/seeding.py
src/nnx/streaming.py
src/nnx/tasks.py
src/nnx/transforms.py
src/nnx/utils.py
src/nnx/vis_utils.py
src/nnx/decisions/__init__.py
src/nnx/decisions/benchmark.py
src/nnx/decisions/jev.py
src/nnx/decisions/jobs.py
src/nnx/decisions/nli.py
src/nnx/decisions/providers.py
src/nnx/decisions/schema.py
src/nnx/diffusion/__init__.py
src/nnx/diffusion/nets.py
src/nnx/diffusion/objective.py
src/nnx/diffusion/sampling.py
src/nnx/diffusion/schedules.py
src/nnx/diffusion/training.py
src/nnx/embeddings/__init__.py
src/nnx/embeddings/contrastive_trainer.py
src/nnx/embeddings/faiss_export.py
src/nnx/finetune/__init__.py
src/nnx/finetune/freezing.py
src/nnx/finetune/loading.py
src/nnx/finetune/param_groups.py
src/nnx/generation/__init__.py
src/nnx/generation/logits_chain.py
src/nnx/generation/logits_processors.py
src/nnx/generation/pipeline.py
src/nnx/generation/sampling.py
src/nnx/interop/__init__.py
src/nnx/interop/ollama.py
src/nnx/interop/gguf/__init__.py
src/nnx/interop/gguf/tensor_name_map.py
src/nnx/interop/gguf/writer.py
src/nnx/nn/__init__.py
src/nnx/nn/callbacks.py
src/nnx/nn/generative_nn_model.py
src/nnx/nn/moe.py
src/nnx/nn/nn_model.py
src/nnx/nn/dataset/__init__.py
src/nnx/nn/dataset/nn_dataset.py
src/nnx/nn/dataset/nn_dataset_base.py
src/nnx/nn/dataset/nn_graph_dataset.py
src/nnx/nn/dataset/nn_preference_dataset.py
src/nnx/nn/dataset/nn_tabular_dataset.py
src/nnx/nn/enum/__init__.py
src/nnx/nn/enum/activations.py
src/nnx/nn/enum/checkpoints.py
src/nnx/nn/enum/devices.py
src/nnx/nn/enum/losses.py
src/nnx/nn/enum/nets.py
src/nnx/nn/enum/optims.py
src/nnx/nn/enum/schedulers.py
src/nnx/nn/net/__init__.py
src/nnx/nn/net/conv_nn.py
src/nnx/nn/net/feed_fwd_moe_nn.py
src/nnx/nn/net/feed_fwd_nn.py
src/nnx/nn/net/graph_att_nn.py
src/nnx/nn/net/graph_conv_nn.py
src/nnx/nn/net/graph_nn_base.py
src/nnx/nn/net/graph_sage_nn.py
src/nnx/nn/net/transformer_layers.py
src/nnx/nn/net/transformer_nn.py
src/nnx/nn/net/vit_nn.py
src/nnx/nn/params/__init__.py
src/nnx/nn/params/_pickling.py
src/nnx/nn/params/nn_checkpoint.py
src/nnx/nn/params/nn_conv_params.py
src/nnx/nn/params/nn_evaluation_data_point.py
src/nnx/nn/params/nn_iteration_data_point.py
src/nnx/nn/params/nn_model_params.py
src/nnx/nn/params/nn_moe_params.py
src/nnx/nn/params/nn_optim_params.py
src/nnx/nn/params/nn_optim_params_builder.py
src/nnx/nn/params/nn_params.py
src/nnx/nn/params/nn_run.py
src/nnx/nn/params/nn_scheduler_params.py
src/nnx/nn/params/nn_scheduler_params_builder.py
src/nnx/nn/params/nn_tokenizer_params.py
src/nnx/nn/params/nn_train_params.py
src/nnx/nn/params/nn_transformer_params.py
src/nnx/nn/params/nn_transformer_params_builder.py
src/nnx/nn/params/nn_vit_params.py
src/nnx/paradigms/__init__.py
src/nnx/paradigms/augmentation.py
src/nnx/paradigms/born_again.py
src/nnx/paradigms/contrastive.py
src/nnx/paradigms/distillation.py
src/nnx/paradigms/dpo.py
src/nnx/paradigms/jepa.py
src/nnx/paradigms/jepa_objective.py
src/nnx/paradigms/moe.py
src/nnx/paradigms/offline_distillation.py
src/nnx/peft/__init__.py
src/nnx/peft/_mode.py
src/nnx/peft/_ownership.py
src/nnx/peft/_source.py
src/nnx/peft/_targets.py
src/nnx/peft/adapters.py
src/nnx/peft/dora.py
src/nnx/peft/ia3.py
src/nnx/peft/lora.py
src/nnx/peft/prefix.py
src/nnx/peft/prompt.py
src/nnx/prune/__init__.py
src/nnx/prune/magnitude.py
src/nnx/prune/semi_structured.py
src/nnx/quantize/__init__.py
src/nnx/quantize/ptq.py
src/nnx/quantize/qat.py
src/nnx/surgery/__init__.py
src/nnx/surgery/_utils.py
src/nnx/surgery/deepen.py
src/nnx/surgery/drop_layer.py
src/nnx/surgery/embedding.py
src/nnx/surgery/low_rank.py
src/nnx/surgery/widen.py
src/nnx/trainer/__init__.py
src/nnx/trainer/params.py
src/nnx/trainer/params_builder.py
src/nnx/trainer/trainer.py
src/nnx/viz/__init__.py
src/nnx/viz/activation.py
src/nnx/viz/attribute.py
src/nnx/viz/gradient_flow.py
src/nnx/viz/netron.py
src/nnx/viz/summary.py
src/nnx/viz/weight_histogram.py
src/thekaveh_nnx.egg-info/PKG-INFO
src/thekaveh_nnx.egg-info/SOURCES.txt
src/thekaveh_nnx.egg-info/dependency_links.txt
src/thekaveh_nnx.egg-info/requires.txt
src/thekaveh_nnx.egg-info/top_level.txt
tests/__init__.py
tests/conftest.py
tests/ddp_scenarios.py
tests/test_abstention.py
tests/test_atomic_write_text.py
tests/test_attempt_provenance.py
tests/test_benchmarking.py
tests/test_builder_branching.py
tests/test_bundle_validation.py
tests/test_calibration.py
tests/test_callbacks.py
tests/test_causal_lm_task.py
tests/test_checkpoint_safetensors.py
tests/test_ci_parallel_suite.py
tests/test_ci_runner_image.py
tests/test_compile_execution.py
tests/test_component_registry.py
tests/test_component_resume.py
tests/test_confusion_matrix.py
tests/test_conv_nn.py
tests/test_core_install.py
tests/test_data_splits.py
tests/test_dataset_batch_sizes.py
tests/test_ddp_adapter.py
tests/test_ddp_resume.py
tests/test_decision_benchmark.py
tests/test_decision_jev.py
tests/test_decision_jobs.py
tests/test_decision_model_pilot.py
tests/test_decision_nli.py
tests/test_decision_nnx_adapter.py
tests/test_decision_schema.py
tests/test_diffusion_nets.py
tests/test_diffusion_objective.py
tests/test_diffusion_sampling.py
tests/test_diffusion_schedules.py
tests/test_diffusion_training.py
tests/test_doc_snippets.py
tests/test_docs_projection.py
tests/test_embeddings_contrastive.py
tests/test_embeddings_faiss_export.py
tests/test_eval_step_fn.py
tests/test_exact_resume.py
tests/test_examples_smoke.py
tests/test_experiment_plan.py
tests/test_export_conformance.py
tests/test_external_module_runtime.py
tests/test_feature_composition.py
tests/test_feed_fwd_moe.py
tests/test_finetune_freezing.py
tests/test_finetune_loading.py
tests/test_finetune_param_groups.py
tests/test_finite_config_validation.py
tests/test_generation_sampling.py
tests/test_generative_nn_model.py
tests/test_graph_classification.py
tests/test_graph_nn_base.py
tests/test_history_journal.py
tests/test_hub_mixin.py
tests/test_imports.py
tests/test_inference_helpers_preserve_training_mode.py
tests/test_integer_config_validation.py
tests/test_interop_gguf_writer.py
tests/test_iteration_data_point.py
tests/test_jepa_encoder_rebuild.py
tests/test_jepa_objective.py
tests/test_legacy_record_pickles.py
tests/test_link_tasks.py
tests/test_logits_chain.py
tests/test_losses.py
tests/test_lr_finder.py
tests/test_metric_accumulators.py
tests/test_metric_specs.py
tests/test_metrics.py
tests/test_model_factories.py
tests/test_model_recipes.py
tests/test_named_monitors.py
tests/test_net_param_compatibility.py
tests/test_nets.py
tests/test_nn_dataset.py
tests/test_nn_graph_dataset.py
tests/test_nn_graph_dataset_full_batch.py
tests/test_nn_moe.py
tests/test_nn_optim_params_builder.py
tests/test_nn_run_repr_html.py
tests/test_nn_scheduler_params_builder.py
tests/test_nn_trainer_params_builder.py
tests/test_nn_transformer_params_builder.py
tests/test_objective_engine.py
tests/test_objective_resume.py
tests/test_offline_distillation.py
tests/test_onnx_dynamo.py
tests/test_optimizer_factories.py
tests/test_optims_adamw.py
tests/test_ordered_logits_pipeline.py
tests/test_paradigms_augmentation.py
tests/test_paradigms_born_again.py
tests/test_paradigms_contrastive.py
tests/test_paradigms_distillation.py
tests/test_paradigms_dpo.py
tests/test_paradigms_feature_kd.py
tests/test_paradigms_jepa.py
tests/test_paradigms_moe.py
tests/test_params_boundary_validation.py
tests/test_params_round_trip.py
tests/test_pass2_d_series.py
tests/test_pass2_f_series.py
tests/test_pass2_n_series.py
tests/test_pass2_ou_series.py
tests/test_pass2_r_series.py
tests/test_pass2_v_series.py
tests/test_peft_adapters.py
tests/test_peft_alias_preflight.py
tests/test_peft_dora.py
tests/test_peft_ia3.py
tests/test_peft_lora.py
tests/test_peft_prefix.py
tests/test_peft_prompt.py
tests/test_per_layer_act_dropout.py
tests/test_phase_tag.py
tests/test_plan_execution.py
tests/test_precision_execution.py
tests/test_precision_policy.py
tests/test_prediction_results.py
tests/test_prediction_stream.py
tests/test_preprocessing.py
tests/test_provenance.py
tests/test_prune_magnitude.py
tests/test_prune_semi_structured.py
tests/test_public_api_exports.py
tests/test_pypi_publication.py
tests/test_quantize_ptq.py
tests/test_quantize_qat.py
tests/test_ranking_task.py
tests/test_release_api_commit.py
tests/test_release_finalize.py
tests/test_reproducible_sdist.py
tests/test_result.py
tests/test_result_boundaries.py
tests/test_resume_point_durability.py
tests/test_resume_point_hashing.py
tests/test_run_bundles.py
tests/test_run_comparison.py
tests/test_run_display_path.py
tests/test_run_load_exact_history.py
tests/test_scheduler_clocks.py
tests/test_schedulers.py
tests/test_search.py
tests/test_step_helpers.py
tests/test_surgery_deepen.py
tests/test_surgery_drop_layer.py
tests/test_surgery_embedding.py
tests/test_surgery_low_rank.py
tests/test_surgery_widen.py
tests/test_task_adapters.py
tests/test_task_evaluation_serialization.py
tests/test_to_onnx_inputs.py
tests/test_tokenizer_params.py
tests/test_train_end_checkpoint.py
tests/test_train_integration.py
tests/test_train_step_hook.py
tests/test_trainer.py
tests/test_trainer_params.py
tests/test_transformer_layers.py
tests/test_transformer_nn.py
tests/test_utils.py
tests/test_vis_utils.py
tests/test_vit_params.py
tests/test_viz_activation.py
tests/test_viz_attribute.py
tests/test_viz_gradient_flow.py
tests/test_viz_netron.py
tests/test_viz_summary.py
tests/test_viz_weight_histogram.py
tests/test_wiki_projection.py
tests/fixtures/release_finalize/after.md
tests/fixtures/release_finalize/before.md
tests/fixtures/run_before_vit/caccfb2e2e30a6de9234b7cfea8fb070/idps.csv
tests/fixtures/run_before_vit/caccfb2e2e30a6de9234b7cfea8fb070/metadata.yaml
tests/fixtures/run_before_vit/caccfb2e2e30a6de9234b7cfea8fb070/run.yaml