vanilla-transformer-jax

View on PyPIReverse Dependencies (0)

0.0.4 vanilla_transformer_jax-0.0.4-py3-none-any.whl

Wheel Details

Project: vanilla-transformer-jax
Version: 0.0.4
Filename: vanilla_transformer_jax-0.0.4-py3-none-any.whl
Download: [link]
Size: 5015
MD5: 1478a026279a7a98c9d7da9367b01bf9
SHA256: 8762fd684a98a58b69a4be3770b0e4bdd59f8e3dbcfe9161a9c4f870520beb8b
Uploaded: 2022-01-19 10:12:27 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: vanilla-transformer-jax
Version: 0.0.4
Summary: JAX/Flax implimentation of 'Attention is All You Need' by Vaswani et al.
Author: bhavnicksm
Author-Email: bhavnicksm[at]gmail.com
Home-Page: https://github.com/Bhavnicksm/vanilla-transformer-jax
Project-Url: Bug Tracker, https://github.com/Bhavnicksm/vanilla-transformer-jax/issues
Keywords: transformer,JAX,Flax,Deep-Learning
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Requires-Dist: jax (>=0.0.0)
Requires-Dist: flax (>=0.0.0)
Requires-Dist: numpy (>=0.0.0)
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 1730 characters]

WHEEL

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

RECORD

Path Digest Size
vtransformer/__init__.py sha256=Jxw21xoaSl9Sl3eVlRJlK60ayrt1kF77pfUfVEWaIyI 71
vtransformer/layers.py sha256=xwAvLIc9B9FRXWXmNfZCcrk5Eqp7iJJXR78fjU_JAXI 4612
vtransformer/model.py sha256=E-4wNscZkfyHuGml8Cm5d2Bu-1I4PW7iE2KpUlASlAw 1800
vanilla_transformer_jax-0.0.4.dist-info/LICENSE sha256=YrXh7p72VA5ghn-UcDTvCbl5wuJ39NlA9OD__xLXvao 1088
vanilla_transformer_jax-0.0.4.dist-info/METADATA sha256=3MtE6ijgX8mOmjL_2oKgODb4kuV5OV3U9cs6Sr_RaJc 2487
vanilla_transformer_jax-0.0.4.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
vanilla_transformer_jax-0.0.4.dist-info/top_level.txt sha256=gv_m0JvCyZGrZWwh3HJYLtnIcLDNwWpQ19EElVFAQF4 13
vanilla_transformer_jax-0.0.4.dist-info/RECORD

top_level.txt

vtransformer