mathformer 1.0.0__tar.gz → 1.0.2__tar.gz

mathformer-1.0.2/MANIFEST.in

@@ -0,0 +1,4 @@
+recursive-include src/mathformer/addformer *
+recursive-include src/mathformer/subformer *
+recursive-include src/mathformer/mulformer *
+recursive-include src/mathformer/divformer *

mathformer-1.0.2/PKG-INFO

@@ -0,0 +1,124 @@
+Metadata-Version: 2.4
+Name: mathformer
+Version: 1.0.2
+Summary: A transformer-based math library
+Author-email: JeremySu0818 <xinghong.su0818@gmail.com>
+Project-URL: Homepage, https://github.com/JeremySu0818/MathFormer
+Project-URL: Bug Tracker, https://github.com/JeremySu0818/MathFormer/issues
+Project-URL: Repository, https://github.com/JeremySu0818/MathFormer
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.0.0
+Requires-Dist: transformers>=4.30.0
+Requires-Dist: safetensors>=0.3.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Dynamic: license-file
+
+# MathFormer
+
+MathFormer is a Python library that leverages Transformer-based language models to perform mathematical operations. Unlike standard arithmetic libraries, MathFormer uses Llama-architecture models to "predict" the results of arithmetic operations, token by token, demonstrating the capability of small language models to learn arithmetic rules.
+
+It supports basic arithmetic operations: **Addition**, **Subtraction**, **Multiplication**, and **Division**.
+
+## Features
+
+- **Transformer-Powered Arithmetic**: Uses specialized Llama-based models for each arithmetic operation.
+- **Large Number Support**: Implements recursive logic to handle multi-digit arithmetic using digit-by-digit prediction (similar to manual calculation).
+- **Unified API**: Easy-to-use functions for `add`, `sub`, `mul`, and `div`.
+- **Resource Management**: Supports lazy loading of models to save memory, and manual unloading.
+- **Custom Tokenizer**: Built-in minimalist tokenizer optimized for mathematical expressions.
+
+## Installation
+
+You can install MathFormer via pip:
+
+```bash
+pip install mathformer
+```
+
+## Quick Start
+
+The simplest way to use MathFormer is through the top-level convenience functions. These functions automatically handle model loading when needed.
+
+```python
+import mathformer
+
+# Addition
+result = mathformer.add(123, 456)
+print(f"123 + 456 = {result}")  # Output: 579
+
+# Subtraction
+result = mathformer.sub(1000, 250)
+print(f"1000 - 250 = {result}") # Output: 750
+
+# Multiplication
+result = mathformer.mul(12, 12)
+print(f"12 * 12 = {result}")    # Output: 144
+
+# Division (returns Quotient and Remainder if applicable)
+result = mathformer.div(100, 3)
+print(f"100 / 3 = {result}")    # Output: Q33R1
+```
+
+You can also pass string expressions:
+
+```python
+print(mathformer.add("100 + 200"))
+print(mathformer.calculate("mul", 50, 4))
+```
+
+## Advanced Usage
+
+For more control over resource usage, you can use the `MathFormerAPI` class directly.
+
+### Managing Resources (Load/Unload)
+
+By default, models are lazy-loaded (loaded only when first requested). You can manually load all models at startup or unload them to free GPU/CPU memory.
+
+```python
+from mathformer import MathFormerAPI
+
+# Initialize API (lazy_load=False to load everything immediately)
+api = MathFormerAPI(lazy_load=True)
+
+# Perform operations
+print(api.add(50, 50))
+
+# Unload all models to free memory
+api.unload_all()
+```
+
+### Context Manager
+
+You can use `MathFormerAPI` as a context manager to ensure models are cleaned up after use:
+
+```python
+from mathformer import MathFormerAPI
+
+with MathFormerAPI() as api:
+    print(api.mul(99, 9))
+# Models are automatically unloaded here
+```
+
+## How It Works
+
+MathFormer isn't just calling Python's `+` or `-` operators. It actually uses a neural network to predict the result!
+
+1.  **Single-Step Prediction**: For small single-digit operations (e.g., `5+7`), it queries a Transformer model customized for that operation.
+2.  **Multi-Digit Logic**: For larger numbers (e.g., `123+456`), the library implements the standard grade-school algorithms (carrying, borrowing, partial products) but delegates the fundamental single-digit arithmetic steps to the Transformer model.
+
+## Requirements
+
+- Python >= 3.8
+- PyTorch >= 2.0.0
+- Transformers >= 4.30.0
+
+## License
+
+This project is licensed under the MIT License.

mathformer-1.0.2/README.md

@@ -0,0 +1,102 @@
+# MathFormer
+
+MathFormer is a Python library that leverages Transformer-based language models to perform mathematical operations. Unlike standard arithmetic libraries, MathFormer uses Llama-architecture models to "predict" the results of arithmetic operations, token by token, demonstrating the capability of small language models to learn arithmetic rules.
+
+It supports basic arithmetic operations: **Addition**, **Subtraction**, **Multiplication**, and **Division**.
+
+## Features
+
+- **Transformer-Powered Arithmetic**: Uses specialized Llama-based models for each arithmetic operation.
+- **Large Number Support**: Implements recursive logic to handle multi-digit arithmetic using digit-by-digit prediction (similar to manual calculation).
+- **Unified API**: Easy-to-use functions for `add`, `sub`, `mul`, and `div`.
+- **Resource Management**: Supports lazy loading of models to save memory, and manual unloading.
+- **Custom Tokenizer**: Built-in minimalist tokenizer optimized for mathematical expressions.
+
+## Installation
+
+You can install MathFormer via pip:
+
+```bash
+pip install mathformer
+```
+
+## Quick Start
+
+The simplest way to use MathFormer is through the top-level convenience functions. These functions automatically handle model loading when needed.
+
+```python
+import mathformer
+
+# Addition
+result = mathformer.add(123, 456)
+print(f"123 + 456 = {result}")  # Output: 579
+
+# Subtraction
+result = mathformer.sub(1000, 250)
+print(f"1000 - 250 = {result}") # Output: 750
+
+# Multiplication
+result = mathformer.mul(12, 12)
+print(f"12 * 12 = {result}")    # Output: 144
+
+# Division (returns Quotient and Remainder if applicable)
+result = mathformer.div(100, 3)
+print(f"100 / 3 = {result}")    # Output: Q33R1
+```
+
+You can also pass string expressions:
+
+```python
+print(mathformer.add("100 + 200"))
+print(mathformer.calculate("mul", 50, 4))
+```
+
+## Advanced Usage
+
+For more control over resource usage, you can use the `MathFormerAPI` class directly.
+
+### Managing Resources (Load/Unload)
+
+By default, models are lazy-loaded (loaded only when first requested). You can manually load all models at startup or unload them to free GPU/CPU memory.
+
+```python
+from mathformer import MathFormerAPI
+
+# Initialize API (lazy_load=False to load everything immediately)
+api = MathFormerAPI(lazy_load=True)
+
+# Perform operations
+print(api.add(50, 50))
+
+# Unload all models to free memory
+api.unload_all()
+```
+
+### Context Manager
+
+You can use `MathFormerAPI` as a context manager to ensure models are cleaned up after use:
+
+```python
+from mathformer import MathFormerAPI
+
+with MathFormerAPI() as api:
+    print(api.mul(99, 9))
+# Models are automatically unloaded here
+```
+
+## How It Works
+
+MathFormer isn't just calling Python's `+` or `-` operators. It actually uses a neural network to predict the result!
+
+1.  **Single-Step Prediction**: For small single-digit operations (e.g., `5+7`), it queries a Transformer model customized for that operation.
+2.  **Multi-Digit Logic**: For larger numbers (e.g., `123+456`), the library implements the standard grade-school algorithms (carrying, borrowing, partial products) but delegates the fundamental single-digit arithmetic steps to the Transformer model.
+
+## Requirements
+
+- Python >= 3.8
+- PyTorch >= 2.0.0
+- Transformers >= 4.30.0
+
+## License
+
+This project is licensed under the MIT License.

{mathformer-1.0.0 → mathformer-1.0.2}/pyproject.toml

@@ -5,13 +5,19 @@
 [tool.setuptools.packages.find]
 where = ["src"]
 
+[tool.setuptools]
+include-package-data = true
+
+[tool.setuptools.package-data]
+mathformer = ["**"]
+
 [build-system]
 requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 
 [project]
 name = "mathformer"
-version = "1.0.0"
+version = "1.0.2"
 description = "A transformer-based math library"
 readme = "README.md"
 authors = [
@@ -36,9 +42,9 @@ dev = [
 ]
 
 [project.urls]
-"Homepage" = "https://github.com/JeremySu0818/MathFormer-API"
-"Bug Tracker" = "https://github.com/JeremySu0818/MathFormer-API/issues"
-"Repository" = "https://github.com/JeremySu0818/MathFormer-API"
+"Homepage" = "https://github.com/JeremySu0818/MathFormer"
+"Bug Tracker" = "https://github.com/JeremySu0818/MathFormer/issues"
+"Repository" = "https://github.com/JeremySu0818/MathFormer"
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]

{mathformer-1.0.0 → mathformer-1.0.2}/src/mathformer/__init__.py

@@ -3,7 +3,7 @@ from .api import MathFormerAPI, MathFormer
 from .tokenizer import MathTokenizer
 
 
-__version__ = "1.0.0"
+__version__ = "1.0.2"
 
 
 _default_api = MathFormerAPI()

mathformer-1.0.2/src/mathformer/addformer/config.json

@@ -0,0 +1,33 @@
+{
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "dtype": "float32",
+  "eos_token_id": 2,
+  "head_dim": 8,
+  "hidden_act": "silu",
+  "hidden_dropout": 0.0,
+  "hidden_size": 64,
+  "initializer_range": 0.02,
+  "intermediate_size": 256,
+  "max_position_embeddings": 64,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 8,
+  "num_key_value_heads": 8,
+  "pad_token_id": 0,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-06,
+  "rope_parameters": {
+    "rope_theta": 10000.0,
+    "rope_type": "default"
+  },
+  "tie_word_embeddings": false,
+  "transformers_version": "5.0.0",
+  "use_cache": false,
+  "vocab_size": 27
+}

mathformer-1.0.2/src/mathformer/addformer/generation_config.json

@@ -0,0 +1,10 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "output_attentions": false,
+  "output_hidden_states": false,
+  "pad_token_id": 0,
+  "transformers_version": "5.0.0",
+  "use_cache": true
+}

mathformer-1.0.2/src/mathformer/addformer/special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/addformer/tokenizer_config.json

@@ -0,0 +1,9 @@
+{
+  "tokenizer_class": "MathTokenizer",
+  "model_max_length": 64,
+  "padding_side": "right",
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/addformer/vocab.json

@@ -0,0 +1,29 @@
+{
+  "<pad>": 0,
+  "<s>": 1,
+  "</s>": 2,
+  "<unk>": 3,
+  "0": 4,
+  "1": 5,
+  "2": 6,
+  "3": 7,
+  "4": 8,
+  "5": 9,
+  "6": 10,
+  "7": 11,
+  "8": 12,
+  "9": 13,
+  "+": 14,
+  "-": 15,
+  "*": 16,
+  "/": 17,
+  "=": 18,
+  ".": 19,
+  "(": 20,
+  ")": 21,
+  "^": 22,
+  "%": 23,
+  " ": 24,
+  "Q": 25,
+  "R": 26
+}

mathformer-1.0.2/src/mathformer/divformer/config.json

@@ -0,0 +1,33 @@
+{
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "dtype": "float32",
+  "eos_token_id": 2,
+  "head_dim": 8,
+  "hidden_act": "silu",
+  "hidden_dropout": 0.0,
+  "hidden_size": 64,
+  "initializer_range": 0.02,
+  "intermediate_size": 256,
+  "max_position_embeddings": 64,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 8,
+  "num_key_value_heads": 8,
+  "pad_token_id": 0,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-06,
+  "rope_parameters": {
+    "rope_theta": 10000.0,
+    "rope_type": "default"
+  },
+  "tie_word_embeddings": false,
+  "transformers_version": "5.0.0",
+  "use_cache": false,
+  "vocab_size": 27
+}

mathformer-1.0.2/src/mathformer/divformer/generation_config.json

@@ -0,0 +1,10 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "output_attentions": false,
+  "output_hidden_states": false,
+  "pad_token_id": 0,
+  "transformers_version": "5.0.0",
+  "use_cache": true
+}

mathformer-1.0.2/src/mathformer/divformer/special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/divformer/tokenizer_config.json

@@ -0,0 +1,9 @@
+{
+  "tokenizer_class": "MathTokenizer",
+  "model_max_length": 64,
+  "padding_side": "right",
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/divformer/vocab.json

@@ -0,0 +1,29 @@
+{
+  "<pad>": 0,
+  "<s>": 1,
+  "</s>": 2,
+  "<unk>": 3,
+  "0": 4,
+  "1": 5,
+  "2": 6,
+  "3": 7,
+  "4": 8,
+  "5": 9,
+  "6": 10,
+  "7": 11,
+  "8": 12,
+  "9": 13,
+  "+": 14,
+  "-": 15,
+  "*": 16,
+  "/": 17,
+  "=": 18,
+  ".": 19,
+  "(": 20,
+  ")": 21,
+  "^": 22,
+  "%": 23,
+  " ": 24,
+  "Q": 25,
+  "R": 26
+}

mathformer-1.0.2/src/mathformer/mulformer/config.json

@@ -0,0 +1,33 @@
+{
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "dtype": "float32",
+  "eos_token_id": 2,
+  "head_dim": 8,
+  "hidden_act": "silu",
+  "hidden_dropout": 0.0,
+  "hidden_size": 64,
+  "initializer_range": 0.02,
+  "intermediate_size": 256,
+  "max_position_embeddings": 64,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 8,
+  "num_key_value_heads": 8,
+  "pad_token_id": 0,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-06,
+  "rope_parameters": {
+    "rope_theta": 10000.0,
+    "rope_type": "default"
+  },
+  "tie_word_embeddings": false,
+  "transformers_version": "5.0.0",
+  "use_cache": false,
+  "vocab_size": 27
+}

mathformer-1.0.2/src/mathformer/mulformer/generation_config.json

@@ -0,0 +1,10 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "output_attentions": false,
+  "output_hidden_states": false,
+  "pad_token_id": 0,
+  "transformers_version": "5.0.0",
+  "use_cache": true
+}

mathformer-1.0.2/src/mathformer/mulformer/special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/mulformer/tokenizer_config.json

@@ -0,0 +1,9 @@
+{
+  "tokenizer_class": "MathTokenizer",
+  "model_max_length": 64,
+  "padding_side": "right",
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/mulformer/vocab.json

@@ -0,0 +1,29 @@
+{
+  "<pad>": 0,
+  "<s>": 1,
+  "</s>": 2,
+  "<unk>": 3,
+  "0": 4,
+  "1": 5,
+  "2": 6,
+  "3": 7,
+  "4": 8,
+  "5": 9,
+  "6": 10,
+  "7": 11,
+  "8": 12,
+  "9": 13,
+  "+": 14,
+  "-": 15,
+  "*": 16,
+  "/": 17,
+  "=": 18,
+  ".": 19,
+  "(": 20,
+  ")": 21,
+  "^": 22,
+  "%": 23,
+  " ": 24,
+  "Q": 25,
+  "R": 26
+}

mathformer-1.0.2/src/mathformer/subformer/config.json

@@ -0,0 +1,33 @@
+{
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "dtype": "float32",
+  "eos_token_id": 2,
+  "head_dim": 8,
+  "hidden_act": "silu",
+  "hidden_dropout": 0.0,
+  "hidden_size": 64,
+  "initializer_range": 0.02,
+  "intermediate_size": 256,
+  "max_position_embeddings": 64,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 8,
+  "num_key_value_heads": 8,
+  "pad_token_id": 0,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-06,
+  "rope_parameters": {
+    "rope_theta": 10000.0,
+    "rope_type": "default"
+  },
+  "tie_word_embeddings": false,
+  "transformers_version": "5.0.0",
+  "use_cache": false,
+  "vocab_size": 27
+}

mathformer-1.0.2/src/mathformer/subformer/generation_config.json

@@ -0,0 +1,10 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "output_attentions": false,
+  "output_hidden_states": false,
+  "pad_token_id": 0,
+  "transformers_version": "5.0.0",
+  "use_cache": true
+}

mathformer-1.0.2/src/mathformer/subformer/special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/subformer/tokenizer_config.json

@@ -0,0 +1,9 @@
+{
+  "tokenizer_class": "MathTokenizer",
+  "model_max_length": 64,
+  "padding_side": "right",
+  "bos_token": "<s>",
+  "eos_token": "</s>",
+  "unk_token": "<unk>",
+  "pad_token": "<pad>"
+}

mathformer-1.0.2/src/mathformer/subformer/vocab.json

@@ -0,0 +1,29 @@
+{
+  "<pad>": 0,
+  "<s>": 1,
+  "</s>": 2,
+  "<unk>": 3,
+  "0": 4,
+  "1": 5,
+  "2": 6,
+  "3": 7,
+  "4": 8,
+  "5": 9,
+  "6": 10,
+  "7": 11,
+  "8": 12,
+  "9": 13,
+  "+": 14,
+  "-": 15,
+  "*": 16,
+  "/": 17,
+  "=": 18,
+  ".": 19,
+  "(": 20,
+  ")": 21,
+  "^": 22,
+  "%": 23,
+  " ": 24,
+  "Q": 25,
+  "R": 26
+}

mathformer-1.0.2/src/mathformer.egg-info/PKG-INFO

@@ -0,0 +1,124 @@
+Metadata-Version: 2.4
+Name: mathformer
+Version: 1.0.2
+Summary: A transformer-based math library
+Author-email: JeremySu0818 <xinghong.su0818@gmail.com>
+Project-URL: Homepage, https://github.com/JeremySu0818/MathFormer
+Project-URL: Bug Tracker, https://github.com/JeremySu0818/MathFormer/issues
+Project-URL: Repository, https://github.com/JeremySu0818/MathFormer
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.0.0
+Requires-Dist: transformers>=4.30.0
+Requires-Dist: safetensors>=0.3.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Dynamic: license-file
+
+# MathFormer
+
+MathFormer is a Python library that leverages Transformer-based language models to perform mathematical operations. Unlike standard arithmetic libraries, MathFormer uses Llama-architecture models to "predict" the results of arithmetic operations, token by token, demonstrating the capability of small language models to learn arithmetic rules.
+
+It supports basic arithmetic operations: **Addition**, **Subtraction**, **Multiplication**, and **Division**.
+
+## Features
+
+- **Transformer-Powered Arithmetic**: Uses specialized Llama-based models for each arithmetic operation.
+- **Large Number Support**: Implements recursive logic to handle multi-digit arithmetic using digit-by-digit prediction (similar to manual calculation).
+- **Unified API**: Easy-to-use functions for `add`, `sub`, `mul`, and `div`.
+- **Resource Management**: Supports lazy loading of models to save memory, and manual unloading.
+- **Custom Tokenizer**: Built-in minimalist tokenizer optimized for mathematical expressions.
+
+## Installation
+
+You can install MathFormer via pip:
+
+```bash
+pip install mathformer
+```
+
+## Quick Start
+
+The simplest way to use MathFormer is through the top-level convenience functions. These functions automatically handle model loading when needed.
+
+```python
+import mathformer
+
+# Addition
+result = mathformer.add(123, 456)
+print(f"123 + 456 = {result}")  # Output: 579
+
+# Subtraction
+result = mathformer.sub(1000, 250)
+print(f"1000 - 250 = {result}") # Output: 750
+
+# Multiplication
+result = mathformer.mul(12, 12)
+print(f"12 * 12 = {result}")    # Output: 144
+
+# Division (returns Quotient and Remainder if applicable)
+result = mathformer.div(100, 3)
+print(f"100 / 3 = {result}")    # Output: Q33R1
+```
+
+You can also pass string expressions:
+
+```python
+print(mathformer.add("100 + 200"))
+print(mathformer.calculate("mul", 50, 4))
+```
+
+## Advanced Usage
+
+For more control over resource usage, you can use the `MathFormerAPI` class directly.
+
+### Managing Resources (Load/Unload)
+
+By default, models are lazy-loaded (loaded only when first requested). You can manually load all models at startup or unload them to free GPU/CPU memory.
+
+```python
+from mathformer import MathFormerAPI
+
+# Initialize API (lazy_load=False to load everything immediately)
+api = MathFormerAPI(lazy_load=True)
+
+# Perform operations
+print(api.add(50, 50))
+
+# Unload all models to free memory
+api.unload_all()
+```
+
+### Context Manager
+
+You can use `MathFormerAPI` as a context manager to ensure models are cleaned up after use:
+
+```python
+from mathformer import MathFormerAPI
+
+with MathFormerAPI() as api:
+    print(api.mul(99, 9))
+# Models are automatically unloaded here
+```
+
+## How It Works
+
+MathFormer isn't just calling Python's `+` or `-` operators. It actually uses a neural network to predict the result!
+
+1.  **Single-Step Prediction**: For small single-digit operations (e.g., `5+7`), it queries a Transformer model customized for that operation.
+2.  **Multi-Digit Logic**: For larger numbers (e.g., `123+456`), the library implements the standard grade-school algorithms (carrying, borrowing, partial products) but delegates the fundamental single-digit arithmetic steps to the Transformer model.
+
+## Requirements
+
+- Python >= 3.8
+- PyTorch >= 2.0.0
+- Transformers >= 4.30.0
+
+## License
+
+This project is licensed under the MIT License.

mathformer-1.0.2/src/mathformer.egg-info/SOURCES.txt

@@ -0,0 +1,44 @@
+LICENSE
+MANIFEST.in
+README.md
+pyproject.toml
+src/mathformer/__init__.py
+src/mathformer/api.py
+src/mathformer/tokenizer.py
+src/mathformer.egg-info/PKG-INFO
+src/mathformer.egg-info/SOURCES.txt
+src/mathformer.egg-info/dependency_links.txt
+src/mathformer.egg-info/requires.txt
+src/mathformer.egg-info/top_level.txt
+src/mathformer/__pycache__/__init__.cpython-311.pyc
+src/mathformer/__pycache__/api.cpython-311.pyc
+src/mathformer/__pycache__/tokenizer.cpython-311.pyc
+src/mathformer/addformer/config.json
+src/mathformer/addformer/generation_config.json
+src/mathformer/addformer/model.safetensors
+src/mathformer/addformer/special_tokens_map.json
+src/mathformer/addformer/tokenizer_config.json
+src/mathformer/addformer/training_args.bin
+src/mathformer/addformer/vocab.json
+src/mathformer/divformer/config.json
+src/mathformer/divformer/generation_config.json
+src/mathformer/divformer/model.safetensors
+src/mathformer/divformer/special_tokens_map.json
+src/mathformer/divformer/tokenizer_config.json
+src/mathformer/divformer/training_args.bin
+src/mathformer/divformer/vocab.json
+src/mathformer/mulformer/config.json
+src/mathformer/mulformer/generation_config.json
+src/mathformer/mulformer/model.safetensors
+src/mathformer/mulformer/special_tokens_map.json
+src/mathformer/mulformer/tokenizer_config.json
+src/mathformer/mulformer/training_args.bin
+src/mathformer/mulformer/vocab.json
+src/mathformer/subformer/config.json
+src/mathformer/subformer/generation_config.json
+src/mathformer/subformer/model.safetensors
+src/mathformer/subformer/special_tokens_map.json
+src/mathformer/subformer/tokenizer_config.json
+src/mathformer/subformer/training_args.bin
+src/mathformer/subformer/vocab.json
+tests/test_api.py

mathformer-1.0.0/PKG-INFO

@@ -1,40 +0,0 @@
-Metadata-Version: 2.4
-Name: mathformer
-Version: 1.0.0
-Summary: A transformer-based math library
-Author-email: JeremySu0818 <xinghong.su0818@gmail.com>
-Project-URL: Homepage, https://github.com/JeremySu0818/MathFormer-API
-Project-URL: Bug Tracker, https://github.com/JeremySu0818/MathFormer-API/issues
-Project-URL: Repository, https://github.com/JeremySu0818/MathFormer-API
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: torch>=2.0.0
-Requires-Dist: transformers>=4.30.0
-Requires-Dist: safetensors>=0.3.0
-Provides-Extra: dev
-Requires-Dist: pytest>=7.0.0; extra == "dev"
-Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
-Dynamic: license-file
-
-# MathFormer
-
-MathFormer is a Python library for mathematical operations using transformer architectures.
-
-## Installation
-
-```bash
-pip install mathformer
-```
-
-## Usage
-
-```python
-import mathformer
-
-# Example usage
-# mathformer.do_something()
-```

mathformer-1.0.0/README.md

@@ -1,18 +0,0 @@
-# MathFormer
-
-MathFormer is a Python library for mathematical operations using transformer architectures.
-
-## Installation
-
-```bash
-pip install mathformer
-```
-
-## Usage
-
-```python
-import mathformer
-
-# Example usage
-# mathformer.do_something()
-```

mathformer-1.0.0/src/mathformer.egg-info/PKG-INFO

@@ -1,40 +0,0 @@
-Metadata-Version: 2.4
-Name: mathformer
-Version: 1.0.0
-Summary: A transformer-based math library
-Author-email: JeremySu0818 <xinghong.su0818@gmail.com>
-Project-URL: Homepage, https://github.com/JeremySu0818/MathFormer-API
-Project-URL: Bug Tracker, https://github.com/JeremySu0818/MathFormer-API/issues
-Project-URL: Repository, https://github.com/JeremySu0818/MathFormer-API
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: torch>=2.0.0
-Requires-Dist: transformers>=4.30.0
-Requires-Dist: safetensors>=0.3.0
-Provides-Extra: dev
-Requires-Dist: pytest>=7.0.0; extra == "dev"
-Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
-Dynamic: license-file
-
-# MathFormer
-
-MathFormer is a Python library for mathematical operations using transformer architectures.
-
-## Installation
-
-```bash
-pip install mathformer
-```
-
-## Usage
-
-```python
-import mathformer
-
-# Example usage
-# mathformer.do_something()
-```

mathformer-1.0.0/src/mathformer.egg-info/SOURCES.txt

@@ -1,12 +0,0 @@
-LICENSE
-README.md
-pyproject.toml
-src/mathformer/__init__.py
-src/mathformer/api.py
-src/mathformer/tokenizer.py
-src/mathformer.egg-info/PKG-INFO
-src/mathformer.egg-info/SOURCES.txt
-src/mathformer.egg-info/dependency_links.txt
-src/mathformer.egg-info/requires.txt
-src/mathformer.egg-info/top_level.txt
-tests/test_api.py