sympy2jax 0.0.5__py3-none-any.whl → 0.0.6__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.
Files changed (6) hide show
  1. sympy2jax/sympy_module.py +1 -1
  2. {sympy2jax-0.0.5.dist-info → sympy2jax-0.0.6.dist-info}/METADATA +18 -27
  3. sympy2jax-0.0.6.dist-info/RECORD +6 -0
  4. {sympy2jax-0.0.5.dist-info → sympy2jax-0.0.6.dist-info}/WHEEL +1 -1
  5. sympy2jax-0.0.5.dist-info/RECORD +0 -6
  6. {sympy2jax-0.0.5.dist-info → sympy2jax-0.0.6.dist-info}/licenses/LICENSE +0 -0
sympy2jax/sympy_module.py CHANGED
@@ -129,7 +129,7 @@ class _Symbol(_AbstractNode):
129
129
  _name: str
130
130
 
131
131
  def __init__(self, expr: sympy.Expr):
132
- self._name = expr.name # pyright: ignore
132
+ self._name = str(expr.name) # pyright: ignore
133
133
 
134
134
  def __call__(self, memodict: dict):
135
135
  try:
{sympy2jax-0.0.5.dist-info → sympy2jax-0.0.6.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.4
2
2
  Name: sympy2jax
3
- Version: 0.0.5
3
+ Version: 0.0.6
4
4
  Summary: Turn SymPy expressions into trainable JAX expressions.
5
5
  Project-URL: repository, https://github.com/google/sympy2jax
6
6
  Author-email: Patrick Kidger <contact@kidger.site>
@@ -274,32 +274,23 @@ Instances have a `.sympy()` method that translates the module back into a PyTree
274
274
 
275
275
  (That's literally the entire documentation, it's super easy.)
276
276
 
277
- ## Finally
277
+ ## See also: other libraries in the JAX ecosystem
278
278
 
279
- ### See also: other libraries in the JAX ecosystem
279
+ **Always useful**
280
+ [Equinox](https://github.com/patrick-kidger/equinox): neural networks and everything not already in core JAX!
281
+ [jaxtyping](https://github.com/patrick-kidger/jaxtyping): type annotations for shape/dtype of arrays.
280
282
 
281
- [jaxtyping](https://github.com/google/jaxtyping): type annotations for shape/dtype of arrays.
283
+ **Deep learning**
284
+ [Optax](https://github.com/deepmind/optax): first-order gradient (SGD, Adam, ...) optimisers.
285
+ [Orbax](https://github.com/google/orbax): checkpointing (async/multi-host/multi-device).
286
+ [Levanter](https://github.com/stanford-crfm/levanter): scalable+reliable training of foundation models (e.g. LLMs).
282
287
 
283
- [Equinox](https://github.com/patrick-kidger/equinox): neural networks.
288
+ **Scientific computing**
289
+ [Diffrax](https://github.com/patrick-kidger/diffrax): numerical differential equation solvers.
290
+ [Optimistix](https://github.com/patrick-kidger/optimistix): root finding, minimisation, fixed points, and least squares.
291
+ [Lineax](https://github.com/patrick-kidger/lineax): linear solvers.
292
+ [BlackJAX](https://github.com/blackjax-devs/blackjax): probabilistic+Bayesian sampling.
293
+ [PySR](https://github.com/milesCranmer/PySR): symbolic regression. (Non-JAX honourable mention!)
284
294
 
285
- [Optax](https://github.com/deepmind/optax): first-order gradient (SGD, Adam, ...) optimisers.
286
-
287
- [Diffrax](https://github.com/patrick-kidger/diffrax): numerical differential equation solvers.
288
-
289
- [Optimistix](https://github.com/patrick-kidger/optimistix): root finding, minimisation, fixed points, and least squares.
290
-
291
- [Lineax](https://github.com/google/lineax): linear solvers.
292
-
293
- [BlackJAX](https://github.com/blackjax-devs/blackjax): probabilistic+Bayesian sampling.
294
-
295
- [Orbax](https://github.com/google/orbax): checkpointing (async/multi-host/multi-device).
296
-
297
- [Eqxvision](https://github.com/paganpasta/eqxvision): computer vision models.
298
-
299
- [Levanter](https://github.com/stanford-crfm/levanter): scalable+reliable training of foundation models (e.g. LLMs).
300
-
301
- [PySR](https://github.com/milesCranmer/PySR): symbolic regression. (Non-JAX honourable mention!)
302
-
303
- ### Disclaimer
304
-
305
- This is not an official Google product.
295
+ **Awesome JAX**
296
+ [Awesome JAX](https://github.com/n2cholas/awesome-jax): a longer list of other JAX projects.
sympy2jax-0.0.6.dist-info/RECORD ADDED
@@ -0,0 +1,6 @@
1
+ sympy2jax/__init__.py,sha256=KqjLNlIiDATWsPhjPLpG-Hud4HMMuq_DP6sISDWVYQg,720
2
+ sympy2jax/sympy_module.py,sha256=zQwxSeW2GbNWQbvqwYaDq_IFx3gVw0Jwe5d6h_YDJO4,9383
3
+ sympy2jax-0.0.6.dist-info/METADATA,sha256=RG9im4iI-5vriRimeDW8wH3aAiy7V1Gcx2FLOpHfsYk,16559
4
+ sympy2jax-0.0.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
5
+ sympy2jax-0.0.6.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
6
+ sympy2jax-0.0.6.dist-info/RECORD,,
{sympy2jax-0.0.5.dist-info → sympy2jax-0.0.6.dist-info}/WHEEL RENAMED
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: hatchling 1.18.0
2
+ Generator: hatchling 1.27.0
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
sympy2jax-0.0.5.dist-info/RECORD DELETED
@@ -1,6 +0,0 @@
1
- sympy2jax/__init__.py,sha256=KqjLNlIiDATWsPhjPLpG-Hud4HMMuq_DP6sISDWVYQg,720
2
- sympy2jax/sympy_module.py,sha256=t1Z8yTRCDgDY_CcG8TF19aAJivKSdz__Ej_OdQ0272g,9378
3
- sympy2jax-0.0.5.dist-info/METADATA,sha256=iJitCmILWFrGKzgHMNxSIUKeX20FrD4alddTdNNHrKc,16459
4
- sympy2jax-0.0.5.dist-info/WHEEL,sha256=9QBuHhg6FNW7lppboF2vKVbCGTVzsFykgRQjjlajrhA,87
5
- sympy2jax-0.0.5.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
6
- sympy2jax-0.0.5.dist-info/RECORD,,