lalamo 0.5.13__py3-none-any.whl → 0.5.14__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
lalamo/__init__.py CHANGED
@@ -15,7 +15,7 @@ from lalamo.speculator import (
15
15
  SpeculatorTrainingEvent,
16
16
  )
17
17
 
18
- __version__ = "0.5.13"
18
+ __version__ = "0.5.14"
19
19
 
20
20
  __all__ = [
21
21
  "AssistantMessage",
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lalamo
3
- Version: 0.5.13
3
+ Version: 0.5.14
4
4
  Summary: JAX library for optimization and export of models for use with the UZU inference engine.
5
5
  Requires-Python: <4,>=3.12
6
6
  Description-Content-Type: text/markdown
@@ -10,7 +10,7 @@ Requires-Dist: click>=8.1.8
10
10
  Requires-Dist: einops>=0.8.0
11
11
  Requires-Dist: equinox>=0.11.11
12
12
  Requires-Dist: huggingface-hub[hf-transfer]>=0.27.1
13
- Requires-Dist: jax>=0.7.2
13
+ Requires-Dist: jax>=0.8.1
14
14
  Requires-Dist: jaxtyping>=0.2.36
15
15
  Requires-Dist: jinja2>=3.1.6
16
16
  Requires-Dist: ml-dtypes>=0.5.1
@@ -23,11 +23,11 @@ Requires-Dist: safetensors>=0.6.2
23
23
  Requires-Dist: polars>=1.33.1
24
24
  Requires-Dist: xxhash>=3.5.0
25
25
  Provides-Extra: cpu
26
- Requires-Dist: jax[cpu]>=0.7.2; extra == "cpu"
26
+ Requires-Dist: jax[cpu]>=0.8.1; extra == "cpu"
27
27
  Provides-Extra: cuda
28
- Requires-Dist: jax[cuda]>=0.7.2; extra == "cuda"
28
+ Requires-Dist: jax[cuda]>=0.8.1; extra == "cuda"
29
29
  Provides-Extra: tpu
30
- Requires-Dist: jax[tpu]>=0.7.2; extra == "tpu"
30
+ Requires-Dist: jax[tpu]>=0.8.1; extra == "tpu"
31
31
  Dynamic: license-file
32
32
 
33
33
  <p align="center">
@@ -1,4 +1,4 @@
1
- lalamo/__init__.py,sha256=v15gN5A_xRGFPWJT--n0m7zlTB3ymrwh1e0YBP4ouA0,815
1
+ lalamo/__init__.py,sha256=W4DF7RbQmPfUVbw1GM5kDfJzUeKaqK7po5iod8VLThQ,815
2
2
  lalamo/common.py,sha256=5NUFD26yQgOnEEk3LaQnce8n-VwJxILkEpFesHZhtQU,3820
3
3
  lalamo/main.py,sha256=GgUT7lT48-XQuAEH7qzsDKG8Lx9iBf-sYBIRhZL9q7E,23978
4
4
  lalamo/message_processor.py,sha256=bSUAQg7CemLTnBV4LtPxJBicAalruDCA-JXjkTYPZ8U,5797
@@ -85,9 +85,9 @@ lalamo/speculator/estimator.py,sha256=4D8dPZCWsrpORb7y8pQ6VsiIg1Cblvvxe6gXCoYtcD
85
85
  lalamo/speculator/inference.py,sha256=5GntUgj0HQLeLn3HIHnVX8EEO0EBzmKeP5-_U7kdFAM,3670
86
86
  lalamo/speculator/ngram.py,sha256=95mdfAWhx4d5XOnOwhyhElnvcy6nlUjYhcbJzqDs414,5875
87
87
  lalamo/speculator/utils.py,sha256=0wZoMMIzzk0Q-3zq5H5f-JBplePNHxywndkrNtOJOyo,1697
88
- lalamo-0.5.13.dist-info/licenses/LICENSE,sha256=diHRfjSEJHD1nnEeMIfMRCjR3UERf8bT3eseD6b1ayA,1072
89
- lalamo-0.5.13.dist-info/METADATA,sha256=nnwQZv5AyDrTw31-bg4sefv2UUSWYCQ3dZfA5MiMT8M,3147
90
- lalamo-0.5.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
91
- lalamo-0.5.13.dist-info/entry_points.txt,sha256=qli7qTfnBk5WP10rOGXXEckHMtt-atJMDWd8jN89Uks,43
92
- lalamo-0.5.13.dist-info/top_level.txt,sha256=VHvWL5JN5XRG36NsN_MieJ7EwRihEOrEjyDaTdFJ-aI,7
93
- lalamo-0.5.13.dist-info/RECORD,,
88
+ lalamo-0.5.14.dist-info/licenses/LICENSE,sha256=diHRfjSEJHD1nnEeMIfMRCjR3UERf8bT3eseD6b1ayA,1072
89
+ lalamo-0.5.14.dist-info/METADATA,sha256=-FRZfOuOcMb_6CuqbsXtECUtU4K7nh074cu3oiogueg,3147
90
+ lalamo-0.5.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
91
+ lalamo-0.5.14.dist-info/entry_points.txt,sha256=qli7qTfnBk5WP10rOGXXEckHMtt-atJMDWd8jN89Uks,43
92
+ lalamo-0.5.14.dist-info/top_level.txt,sha256=VHvWL5JN5XRG36NsN_MieJ7EwRihEOrEjyDaTdFJ-aI,7
93
+ lalamo-0.5.14.dist-info/RECORD,,