rxnn 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

{src → rxnn}/experimental/attention.py

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

{src → rxnn}/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.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
+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
 
 
 class RxTAlphaComponentConfig(TypedDict):

{src → rxnn}/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.transformers.models import ReactiveTransformerEncoder, ReactiveTransformerDecoder
-from src.training.base import BaseTrainer
+from src.rxnn.transformers.models import ReactiveTransformerEncoder, ReactiveTransformerDecoder
+from src.rxnn.training.base import BaseTrainer
 
 class MLMHead(nn.Module, PyTorchModelHubMixin, license="apache-2.0"):
     def __init__(self, embed_dim: int, vocab_size: int, *args, **kwargs):

{src → rxnn}/training/callbacks.py

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

{src → rxnn}/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.memory.stm import ShortTermMemory
+from src.rxnn.memory.stm import ShortTermMemory
 
 
 class ReactiveTransformerBase(nn.Module):

{rxnn-0.1.2.dist-info → rxnn-0.1.3.dist-info}/METADATA

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

rxnn-0.1.3.dist-info/RECORD

@@ -0,0 +1,29 @@
+rxnn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rxnn/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rxnn/experimental/attention.py,sha256=z7RkLvgY815JHEfCLq1gtW9yKdumjEJ3yGCrA7jCyRU,5638
+rxnn/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rxnn/memory/norm.py,sha256=Ofl8Q5NYEF9GQeO0bhM43tkTW91J0y6TSvTAOYMgloM,6278
+rxnn/memory/stm.py,sha256=EsD8slSP4_9dLuq6aFPDmuFe8PWilxh90so5Z3nm-ig,2057
+rxnn/rxt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rxnn/rxt/models.py,sha256=M63lz7qN3f3cvxgc4bV_WLgUQCnVRnDU8NH54wE_eks,6967
+rxnn/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rxnn/training/base.py,sha256=DHankUIta0wWuczN1O0pK7MCis9dYUvc5dAFkap9t5Q,11030
+rxnn/training/bml.py,sha256=zX7bvlKyLIPZKYAvt7JVLNZBMz8eSTd3kuucr92S-dc,14589
+rxnn/training/callbacks.py,sha256=8RjFuMCtHwmlXJMp6X68wWTJdZ8xyFykckdmb7FjHNU,21098
+rxnn/training/dataset.py,sha256=vQ5mDF3bA0HXya474n4D4iL8Mn3AEpJukgzFNVkxjGU,5106
+rxnn/training/scheduler.py,sha256=ow6oALzWjWQmHSpcJEjv6tg4g4CDMvr73TypxfcefMc,712
+rxnn/training/tokenizer.py,sha256=4Y41f07uo2KPA_7bp3FCcwGKbXoS2hsckOoXUsXfQxY,8052
+rxnn/transformers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rxnn/transformers/attention.py,sha256=fqNziF9gZOLM-RnrkaCOkel8w-EkLEAOiv7WO5G6IMw,14021
+rxnn/transformers/ff.py,sha256=jJnuBDsnnX5uYC_WZH8cXAYrMnz0P-iX7MwcPivjRtI,2533
+rxnn/transformers/layers.py,sha256=PBlnQzS_cyaAjEAQEgdsttSXhtwWsx4PKxRAtPMhTbY,5361
+rxnn/transformers/mask.py,sha256=J0cfLVLt3SzS2ra3KcY4khrkhI975Dw4CjpUi3Sn25s,419
+rxnn/transformers/models.py,sha256=6QVuUq7VYNniIjGor1YKUy2xK0LWMOFuJzymGVY6Tcc,6721
+rxnn/transformers/moe.py,sha256=JQ5QSX4FS7S-fqB7-s1ZmJbPpOeD_Injn8o4vo7wGQE,4936
+rxnn/transformers/positional.py,sha256=2l38RS0Dini3f6Z3LUHr3XwWzg1UK7fO2C6wazWDAYU,4292
+rxnn/transformers/sampler.py,sha256=wSz_1wNloqtuiix5w2Mcsj5NhaO9QlY0j__TVG7wJnM,3938
+rxnn/utils.py,sha256=d5U8i5ukovgDyqiycc2AoxObTz_eF_bgo2MKvdtJ98s,467
+rxnn-0.1.3.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
+rxnn-0.1.3.dist-info/METADATA,sha256=i50r1jTnBl7NchAb31op-Hprkb8gQhMF8ErXREIvh5E,14486
+rxnn-0.1.3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+rxnn-0.1.3.dist-info/RECORD,,

rxnn-0.1.2.dist-info/RECORD

@@ -1,29 +0,0 @@
-src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/experimental/attention.py,sha256=BxNFOqQcz3v_wzn5n07UuZQIRLkVxNJEqZedvv8K_WQ,5633
-src/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/memory/norm.py,sha256=Ofl8Q5NYEF9GQeO0bhM43tkTW91J0y6TSvTAOYMgloM,6278
-src/memory/stm.py,sha256=EsD8slSP4_9dLuq6aFPDmuFe8PWilxh90so5Z3nm-ig,2057
-src/rxt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/rxt/models.py,sha256=jmxdQ0FHjqYKVrwoEURez1WV-0quv-5q2eBFGhXyJM0,6932
-src/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/training/base.py,sha256=DHankUIta0wWuczN1O0pK7MCis9dYUvc5dAFkap9t5Q,11030
-src/training/bml.py,sha256=Al-qcQYQ-GsdXfa1DB1YF2vGJKgfIfMBy0GlZabUz1Y,14579
-src/training/callbacks.py,sha256=VAjwaAoSgYgU5WV9FybINznLhBRX9WMWNdgS41FtNfY,21093
-src/training/dataset.py,sha256=vQ5mDF3bA0HXya474n4D4iL8Mn3AEpJukgzFNVkxjGU,5106
-src/training/scheduler.py,sha256=ow6oALzWjWQmHSpcJEjv6tg4g4CDMvr73TypxfcefMc,712
-src/training/tokenizer.py,sha256=4Y41f07uo2KPA_7bp3FCcwGKbXoS2hsckOoXUsXfQxY,8052
-src/transformers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/transformers/attention.py,sha256=fqNziF9gZOLM-RnrkaCOkel8w-EkLEAOiv7WO5G6IMw,14021
-src/transformers/ff.py,sha256=jJnuBDsnnX5uYC_WZH8cXAYrMnz0P-iX7MwcPivjRtI,2533
-src/transformers/layers.py,sha256=PBlnQzS_cyaAjEAQEgdsttSXhtwWsx4PKxRAtPMhTbY,5361
-src/transformers/mask.py,sha256=J0cfLVLt3SzS2ra3KcY4khrkhI975Dw4CjpUi3Sn25s,419
-src/transformers/models.py,sha256=aVY3YTKGKvwzRraduhIgO8phPDRkJytdTglDQm_U2_c,6716
-src/transformers/moe.py,sha256=JQ5QSX4FS7S-fqB7-s1ZmJbPpOeD_Injn8o4vo7wGQE,4936
-src/transformers/positional.py,sha256=2l38RS0Dini3f6Z3LUHr3XwWzg1UK7fO2C6wazWDAYU,4292
-src/transformers/sampler.py,sha256=wSz_1wNloqtuiix5w2Mcsj5NhaO9QlY0j__TVG7wJnM,3938
-src/utils.py,sha256=d5U8i5ukovgDyqiycc2AoxObTz_eF_bgo2MKvdtJ98s,467
-rxnn-0.1.2.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
-rxnn-0.1.2.dist-info/METADATA,sha256=N856s1leESK8WagTniy8g-o11VFlkT-PPn9RtcLo42o,14486
-rxnn-0.1.2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-rxnn-0.1.2.dist-info/RECORD,,