jaxkan

View on PyPIReverse Dependencies (0)

0.2.0 jaxkan-0.2.0-py3-none-any.whl

Wheel Details

Project: jaxkan
Version: 0.2.0
Filename: jaxkan-0.2.0-py3-none-any.whl
Download: [link]
Size: 25307
MD5: 7d97936d10f311d29103c20fb7067cb3
SHA256: cd176d547aefd3ad3feff59d3379aaa42749d40c40d68ca8b58290ececbacd1e
Uploaded: 2025-01-07 16:35:46 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jaxkan
Version: 0.2.0
Summary: A JAX implementation of Kolmogorov-Arnold Networks
Author-Email: Spyros Rigas <rigassp[at]gmail.com>, Michalis Papachristou <mixpap[at]phys.uoa.gr>
Project-Url: Repository, https://github.com/srigas/jaxkan
Project-Url: Issues, https://github.com/srigas/jaxkan/issues
License: MIT License Copyright (c) 2024 Spyros Rigas, Michalis Papachristou Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
Keywords: JAX,NNX,KANs,Kolmogorov-Arnold,PIKAN
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: License :: OSI Approved :: MIT License
Requires-Python: >=3.10
Requires-Dist: numpy (==2.1.3)
Requires-Dist: scipy (==1.14.1)
Requires-Dist: jax[cpu] (==0.4.35)
Requires-Dist: optax (==0.2.4)
Requires-Dist: flax (==0.10.2)
Requires-Dist: pytest (==8.3.4)
Requires-Dist: jax[cuda12] (==0.4.35); extra == "gpu"
Requires-Dist: jupyterlab (==4.3.0); extra == "doc"
Requires-Dist: matplotlib (==3.9.2); extra == "doc"
Requires-Dist: scikit-learn (==1.5.2); extra == "doc"
Requires-Dist: sphinx; extra == "doc"
Requires-Dist: sphinx-rtd-theme; extra == "doc"
Requires-Dist: sphinx-autobuild; extra == "doc"
Requires-Dist: myst-parser; extra == "doc"
Requires-Dist: sphinx_design; extra == "doc"
Requires-Dist: sphinx-autodoc-typehints; extra == "doc"
Requires-Dist: nbsphinx; extra == "doc"
Requires-Dist: nbsphinx-link; extra == "doc"
Requires-Dist: pandoc; extra == "doc"
Provides-Extra: gpu
Provides-Extra: doc
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 2083 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.7.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jaxkan/KAN.py sha256=IIBRoOjtNc-jb0cj0K8kc8bCOew8ObVG2bpWMmWrd8U 5169
jaxkan/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxkan/layers/BaseLayer.py sha256=5S14spddG5uYCDL_MU9-9ENAEyfIZwMJGIlP36myi7I 17780
jaxkan/layers/ChebyLayer.py sha256=DYoTnISTbwI2_izYqMTvIddmYmM9B3KI-UYt9CPxExw 5525
jaxkan/layers/FourierLayer.py sha256=iRvkDpPJM12XXmaAt5jwvRvSvPWV0ZTheXrJvl_TPoA 6801
jaxkan/layers/ModifiedChebyLayer.py sha256=AWaU67qYO0bpJINvOZFPVkFMJGSCIijAh9KsW2ztfps 5856
jaxkan/layers/SplineLayer.py sha256=7tRlb_g4Gi4hlglAEle6WCx6CI4fRui1G2PosRuiW3Y 16207
jaxkan/layers/__init__.py sha256=AuaQMSGCwtwNHSBTjdjBLA6M56gfga5_9TmmWKN4Ma4 1075
jaxkan/utils/PIKAN.py sha256=kem5nCm8bur5b_UphnK5onAyseA01H86cK6gjZ7Oc-0 15984
jaxkan/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxkan/utils/general.py sha256=gea1Nc2f0ftqtoQQE3NI4topYghCQVNiammph3EBdGw 5659
jaxkan-0.2.0.dist-info/LICENSE sha256=XEFMXSzQ88q_tL7MypI3G8sS_xytpYCSbMM1aK2HPzU 1092
jaxkan-0.2.0.dist-info/METADATA sha256=IsZTfSN3VWexPMX7KIM6X50IqDEgFVZUeyeY5yICRU8 4927
jaxkan-0.2.0.dist-info/WHEEL sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE 91
jaxkan-0.2.0.dist-info/top_level.txt sha256=xo4FenCxyXxBje0XrqLqstj0TUmTuTUV33ZJiX1Av0k 7
jaxkan-0.2.0.dist-info/RECORD

top_level.txt

jaxkan