Introduce param module and refactor surrogate gradients (#63)

* Add parameter transformation module; implement Param class and related transforms

* Refactor __init__.py; update __all__ exports to include state module

* Add surrogate gradient implementation; create Surrogate class and refactor related modules

* Refactor module exports; update __module__ attributes to 'braintools.param'

* Add new transformation classes: Positive, Negative, ScaledSigmoid, Power, Ordered, Simplex, and UnitVector; implement log_abs_det_jacobian methods

* Update CI configuration; remove develop and release branches from push triggers

* Bump version to 0.1.5; update version_info to reflect new version and add support for Python 3.14 in pyproject.toml

* Remove lint job from CI configuration; simplify testing setup

Author: chaoming0625

.github/workflows/CI.yml

@@ -4,13 +4,8 @@ on:
   push:
     branches:
       - main
-      - develop
-      - 'release/**'
   pull_request:
-    branches:
-      - main
-      - develop
-  workflow_call:
+
 
 permissions:
   contents: read
@@ -20,25 +15,9 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-#  lint:
-#    name: Lint
-#    runs-on: ubuntu-latest
-#    steps:
-#      - uses: actions/checkout@v6
-#      - uses: actions/setup-python@v6
-#        with:
-#          python-version: '3.12'
-#          cache: pip
-#      - name: Install pre-commit
-#        run: |
-#          python -m pip install --upgrade pip
-#          pip install pre-commit
-#      - name: Run pre-commit
-#        run: pre-commit run --all-files --show-diff-on-failure --color always
 
   tests:
     name: Tests (${{ matrix.os }} / py${{ matrix.python }})
-#    needs: lint
     runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false

braintools/__init__.py

@@ -14,15 +14,16 @@
 # ==============================================================================
 
 
-__version__ = "0.1.4"
-__version_info__ = (0, 1, 4)
+__version__ = "0.1.5"
+__version_info__ = tuple(map(int, __version__.split(".")))
 
 from . import conn
 from . import file
 from . import init
 from . import input
 from . import metric
 from . import optim
+from . import param
 from . import quad
 from . import surrogate
 from . import tree
@@ -31,15 +32,12 @@
 from ._spike_encoder import __all__ as encoder_all
 from ._spike_operation import *
 from ._spike_operation import __all__ as operation_all
-from ._transform import *
-from ._transform import __all__ as transform_all
 
 __all__ = [
-    'conn', 'input', 'init',
-    'file', 'metric', 'visualize',
-    'optim', 'tree', 'quad',
-    'surrogate',
-]
-__all__ = __all__ + encoder_all + transform_all + operation_all
+              'conn', 'input', 'init',
+              'file', 'metric', 'visualize',
+              'optim', 'tree', 'quad',
+              'surrogate', 'param',
+          ] + encoder_all + operation_all
 
-del encoder_all, transform_all, operation_all
+del encoder_all, operation_all

braintools/_transform_test.py

@@ -0,0 +1,25 @@
+# Copyright 2025 BrainX Ecosystem Limited. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+from ._base import *
+from ._base import __all__ as base_all
+from ._state import *
+from ._state import __all__ as state_all
+from ._transform import *
+from ._transform import __all__ as transform_all
+
+__all__ = state_all + transform_all + base_all
+
+del transform_all, base_all, state_all