CHANGELOG.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
src/nnx/__init__.py
src/nnx/_metrics.py
src/nnx/_step_helpers.py
src/nnx/lr_finder.py
src/nnx/py.typed
src/nnx/seeding.py
src/nnx/utils.py
src/nnx/vis_utils.py
src/nnx/diffusion/__init__.py
src/nnx/diffusion/nets.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/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/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/nn_checkpoint.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_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/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/moe.py
src/nnx/peft/__init__.py
src/nnx/peft/_source.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/test_callbacks.py
tests/test_checkpoint_safetensors.py
tests/test_confusion_matrix.py
tests/test_diffusion_nets.py
tests/test_diffusion_sampling.py
tests/test_diffusion_schedules.py
tests/test_diffusion_training.py
tests/test_embeddings_contrastive.py
tests/test_embeddings_faiss_export.py
tests/test_finetune_freezing.py
tests/test_finetune_loading.py
tests/test_finetune_param_groups.py
tests/test_generation_sampling.py
tests/test_generative_nn_model.py
tests/test_graph_nn_base.py
tests/test_hub_mixin.py
tests/test_imports.py
tests/test_inference_helpers_preserve_training_mode.py
tests/test_interop_gguf_writer.py
tests/test_iteration_data_point.py
tests/test_logits_chain.py
tests/test_losses.py
tests/test_lr_finder.py
tests/test_metrics.py
tests/test_nets.py
tests/test_nn_dataset.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_onnx_dynamo.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_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_dora.py
tests/test_peft_ia3.py
tests/test_peft_lora.py
tests/test_peft_prefix.py
tests/test_peft_prompt.py
tests/test_phase_tag.py
tests/test_prune_magnitude.py
tests/test_prune_semi_structured.py
tests/test_pypi_publication.py
tests/test_quantize_ptq.py
tests/test_quantize_qat.py
tests/test_schedulers.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_to_onnx_inputs.py
tests/test_tokenizer_params.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_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