Skip to content

Add Python package structure to models directory#9

Draft
Copilot wants to merge 3 commits intomainfrom
copilot/add-tiny-recursive-models
Draft

Add Python package structure to models directory#9
Copilot wants to merge 3 commits intomainfrom
copilot/add-tiny-recursive-models

Conversation

Copy link

Copilot AI commented Jan 31, 2026

The models/ and models/recursive_reasoning/ directories lacked __init__.py files, relying on implicit namespace packages. This PR adds proper package structure.

Changes

  • models/__init__.py: Establishes models as a package, exports core utilities (EMA, layers, losses, sparse embedding)
  • models/recursive_reasoning/__init__.py: Establishes recursive_reasoning subpackage, exports all model variants (TRM, HRM, transformers baseline, ERS_PMLL, etc.)

Implementation Details

Both files define __all__ for API documentation but avoid eager imports to prevent torch dependency loading on package import:

# models/recursive_reasoning/__init__.py
"""Recursive reasoning models for TinyRecursiveModels.

Classes are available via explicit imports:
    from models.recursive_reasoning.trm import TinyRecursiveReasoningModel_ACTV1
"""

__all__ = [
    "TinyRecursiveReasoningModel_ACTV1",
    "HierarchicalReasoningModel_ACTV1",
    # ...
]

Backward compatible with existing dynamic model loading via importlib.import_module() used in config-driven instantiation.

Original prompt

https://github.com/drqsatoshi/TinyRecursiveModels/tasks/7d925690-6100-4dad-9a4b-fa1950629350?session_id=27d34d06-1d4f-47e6-a171-0d6660c65d14


💬 We'd love your input! Share your thoughts on Copilot coding agent in our 2 minute survey.

Copilot AI and others added 2 commits January 31, 2026 04:40
Co-authored-by: drqsatoshi <240532885+drqsatoshi@users.noreply.github.com>
Co-authored-by: drqsatoshi <240532885+drqsatoshi@users.noreply.github.com>
Copilot AI changed the title [WIP] Add tiny recursive models implementation Add Python package structure to models directory Jan 31, 2026
Copilot AI requested a review from drqsatoshi January 31, 2026 04:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants