rxnn 0.1.1__tar.gz → 0.1.2__tar.gz

{rxnn-0.1.1 → rxnn-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rxnn
-Version: 0.1.1
+Version: 0.1.2
 Summary: RxNN: Reactive Neural Networks Platform
 License: Apache-2.0
 Keywords: deep-learning,ai,machine-learning

{rxnn-0.1.1 → rxnn-0.1.2}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rxnn"
-version = "0.1.1"
+version = "0.1.2"
 description = "RxNN: Reactive Neural Networks Platform"
 
 packages = [
-    { include = "rxnn", from = "src" },
+    { include = "src" },
 ]
 
 license = "Apache-2.0"

{rxnn-0.1.1/src/rxnn → rxnn-0.1.2/src}/experimental/attention.py

@@ -1,6 +1,6 @@
 import torch
 from torch import nn
-from src.rxnn.transformers.attention import MultiHeadAttention
+from src.transformers.attention import MultiHeadAttention
 
 class FlexAttention(MultiHeadAttention):
     def __init__(

{rxnn-0.1.1/src/rxnn → rxnn-0.1.2/src}/rxt/models.py

@@ -2,13 +2,13 @@ import torch
 from torch import nn
 from typing import TypedDict, Union
 from huggingface_hub import PyTorchModelHubMixin
-from src.rxnn.transformers.positional import RotaryPositionalEmbedding
-from src.rxnn.transformers.attention import init_attention
-from src.rxnn.transformers.layers import ReactiveTransformerLayer
-from src.rxnn.transformers.models import ReactiveTransformerBase, ReactiveTransformerEncoder, ReactiveTransformerDecoder
-from src.rxnn.transformers.ff import get_activation_layer
-from src.rxnn.memory.stm import ShortTermMemory
-from src.rxnn.utils import get_model_size
+from src.transformers.positional import RotaryPositionalEmbedding
+from src.transformers.attention import init_attention
+from src.transformers.layers import ReactiveTransformerLayer
+from src.transformers.models import ReactiveTransformerBase, ReactiveTransformerEncoder, ReactiveTransformerDecoder
+from src.transformers.ff import get_activation_layer
+from src.memory.stm import ShortTermMemory
+from src.utils import get_model_size
 
 
 class RxTAlphaComponentConfig(TypedDict):

{rxnn-0.1.1/src/rxnn → rxnn-0.1.2/src}/training/bml.py

@@ -5,8 +5,8 @@ import math
 from huggingface_hub import PyTorchModelHubMixin
 from typing import Union
 import torch.distributed as dist
-from src.rxnn.transformers.models import ReactiveTransformerEncoder, ReactiveTransformerDecoder
-from src.rxnn.training.base import BaseTrainer
+from src.transformers.models import ReactiveTransformerEncoder, ReactiveTransformerDecoder
+from src.training.base import BaseTrainer
 
 class MLMHead(nn.Module, PyTorchModelHubMixin, license="apache-2.0"):
     def __init__(self, embed_dim: int, vocab_size: int, *args, **kwargs):

{rxnn-0.1.1/src/rxnn → rxnn-0.1.2/src}/training/callbacks.py

@@ -3,7 +3,7 @@ import numpy as np
 import torch
 import torch.nn as nn
 from typing import Union
-from src.rxnn.utils import human_format
+from src.utils import human_format
 from torch.nn.parallel import DistributedDataParallel
 from huggingface_hub import PyTorchModelHubMixin
 

{rxnn-0.1.1/src/rxnn → rxnn-0.1.2/src}/transformers/models.py

@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 from positional import AbsolutePositionalEmbedding
 from mask import create_causal_mask
-from src.rxnn.memory.stm import ShortTermMemory
+from src.memory.stm import ShortTermMemory
 
 
 class ReactiveTransformerBase(nn.Module):