Wheelodex » Projects » mistral-jax

mistral-jax

View on PyPI — Reverse Dependencies (0)

0.0.3 mistral_jax-0.0.3-py3-none-any.whl

Wheel Details

Project: mistral-jax
Version: 0.0.3
Filename: mistral_jax-0.0.3-py3-none-any.whl
Download: [link]
Size: 22631
MD5: f88d04aa8c7ba487f0ad6ee64f168e5d
SHA256: 48fbf30a08f27dbf27c633e82a579612bdff2932370a4f47ea996bcb716c6283
Uploaded: 2024-03-23 16:50:54 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: mistral-jax
Version: 0.0.3
Summary: JAX implementation of the Mistral model.
Author-Email: yixiaoer <miyaku[at]yixiaoer.sg>
Project-Url: Repository, https://github.com/yixiaoer/mistral-jax
Project-Url: Issues, https://github.com/yixiaoer/mistral-jax/issues
Classifier: Development Status :: 4 - Beta
Classifier: Programming Language :: Python
Classifier: License :: OSI Approved :: MIT License
Requires-Python: >=3.10
Requires-Dist: mypy
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: torch
Requires-Dist: transformers
Requires-Dist: einops
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 15062 characters]

WHEEL

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

RECORD

Path Digest Size
mistral/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mistral/lib/__init__.py sha256=Lzwa87qWgs9PuMC9kj5BAkcQawo7oXP8PpUsCvHVD1c 138
mistral/lib/array_conversion.py sha256=fc5EEg_beNIt_4cxumnpwRGliYMPqqIzcCCPYWVKPoU 576
mistral/lib/collate_fn.py sha256=1RnzAMySHFuOJcNaM2c2SLzk_S0UV_FITTjPLeRmDSE 3185
mistral/lib/einshard.py sha256=TwXNyus6EqSnCFngdG9t604YVMtM0rnNNywyJNJfHpg 8204
mistral/lib/example_data.py sha256=3nHSaQ7oDwJGQ5o1JfJnbmJifs-1Gjk-_KCoYlaitP0 829
mistral/lib/generate.py sha256=nPeUFFyQ6gToVuzNWazrwL-u-cGA-x0d4rUFv4CGkuk 7436
mistral/model/__init__.py sha256=r7djSv2DfhsJsU7P6xeogOgysT6VmXxuQkJcPXdL77w 253
mistral/model/attention.py sha256=V7fVnu-xZStPJY7Ehg7viORwuI0TpJAugDlAni5AN6Y 4855
mistral/model/decoder.py sha256=HoPHd3iTFs12Ok5B4ZtMO1MENTillyFQiolRZ71iATM 1142
mistral/model/decoder_block.py sha256=rJe0r2v0Ouq8tFS5r800clv3Fv8Bsjbu_nmJSxeUHtU 2383
mistral/model/embedding.py sha256=AkcPd8-f8AhDeEuCPjL0pRxhZUAE432w75G5502hOPQ 1125
mistral/model/kvcache.py sha256=QEQIQjURUkgdvstggs7hsxFgEutkd8GImSYIURkzjig 70
mistral/model/mistral_lm.py sha256=6-f09xDMmgPuqWtpgWBFIelfhTJzMDU19RhOebQBKRY 2706
mistral/model/mistral_model.py sha256=gyCwrfNCqb9VKRv2mNuKs4yDh9OYgYJi9wEurG-Gv9A 1654
mistral/model/mlp_layer.py sha256=Bi--PekEW88822RCR71QZfexTaM3RoKav1KY26eaecc 1314
mistral/model/rms_norm.py sha256=mAgpbuyr6nnnrUeotX20q6j8AYR_qJACZqyFRN7Ujrs 1475
mistral/model/rotary_embedding.py sha256=87Y1YfdpFlyn8gVAJXDOFosE3SdXIGy-SUm0_vvW9sg 2239
mistral_jax-0.0.3.dist-info/LICENSE sha256=AWbEMAUVzy0fTxQvntgL0ST6cIBLVeM0IxqbOMQ7iOw 1065
mistral_jax-0.0.3.dist-info/METADATA sha256=E_zN_3zUlxfxv-liQ7S2R1tBq4PtFOdA9AdTwkHl8_k 15708
mistral_jax-0.0.3.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
mistral_jax-0.0.3.dist-info/top_level.txt sha256=jJfoDMgfoPr_wpz3UV8qR8mJALu-Ux2QPAErFOLEO60 8
mistral_jax-0.0.3.dist-info/RECORD — —

top_level.txt

mistral