jaxkan

View on PyPIReverse Dependencies (0)

0.1.8 jaxkan-0.1.8-py3-none-any.whl

Wheel Details

Project: jaxkan
Version: 0.1.8
Filename: jaxkan-0.1.8-py3-none-any.whl
Download: [link]
Size: 20544
MD5: 3c44592b25c81f3c1d6b552717e22791
SHA256: 5db9c5a5e9ca368d902a88cb01e44508158ad63492e100b4daffa5113328598e
Uploaded: 2024-11-28 13:47:49 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jaxkan
Version: 0.1.8
Summary: A JAX-based implementation of Kolmogorov-Arnold Networks
Author: Spyros Rigas, Michalis Papachristou
Author-Email: rigassp[at]gmail.com
Home-Page: https://github.com/srigas/jaxkan
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Requires-Dist: scipy (==1.13.1)
Requires-Dist: numpy (==1.26.4)
Requires-Dist: flax (==0.8.3)
Requires-Dist: jax[cpu] (==0.4.28)
Requires-Dist: optax (==0.2.2)
Requires-Dist: jax[cuda12]; extra == "gpu"
Provides-Extra: gpu
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 1909 characters]

WHEEL

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

RECORD

Path Digest Size
jaxkan/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxkan/bases/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxkan/bases/relus.py sha256=CmiFEqLkoi8bzYJ-w2cche-gpsVL1G2aDEHTuqcmuks 2848
jaxkan/bases/splines.py sha256=gZiX6jO7_ShEwMxKbBUkWf0dAZnZwYVIUHJu4mhLjdc 1577
jaxkan/models/KAN.py sha256=hMxyLskewptEsNrcAepqk0oC-VeymcErkNTA3vhsDRw 5160
jaxkan/models/KANLayer.py sha256=x7AqSRSPgvoqVANYBoOH1l8ocugbg4NbhOuLuRtDLOI 11757
jaxkan/models/ReLUKAN.py sha256=Z6qsOcdAlQ_DxdH7D67DPjadMJCVz3YYy6LyyNLkwc4 5365
jaxkan/models/ReLUKANLayer.py sha256=uOuMKxwR9dEPuakTW4Ndb_xAXl5P1upimqLkHTlPLJM 11818
jaxkan/models/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxkan/utils/PIKAN.py sha256=rK4b0spgQYHnemDOENwu9rK_CjBj9tka3oRu2HlXyUI 15799
jaxkan/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxkan/utils/general.py sha256=FO1GT5ZoukMrqZYeRv45m77bJj8ZIdt5tKC8m5o5jys 1622
jaxkan-0.1.8.dist-info/LICENSE sha256=XEFMXSzQ88q_tL7MypI3G8sS_xytpYCSbMM1aK2HPzU 1092
jaxkan-0.1.8.dist-info/METADATA sha256=tHJwY78-gD--TOrupDoFpOKwlbtwpIzaBKZEj7EwKWQ 2585
jaxkan-0.1.8.dist-info/WHEEL sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U 91
jaxkan-0.1.8.dist-info/top_level.txt sha256=xo4FenCxyXxBje0XrqLqstj0TUmTuTUV33ZJiX1Av0k 7
jaxkan-0.1.8.dist-info/RECORD

top_level.txt

jaxkan