sympy2jax
jaxtyping
| sympy2jax | jaxtyping | |
|---|---|---|
| 1 | 7 | |
| 307 | 987 | |
| -0.3% | 4.7% | |
| 4.4 | 8.3 | |
| about 2 months ago | 10 days ago | |
| Python | Python | |
| Apache License 2.0 | GNU General Public License v3.0 or later |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
sympy2jax
-
Scientific computing in JAX
sympy2jax: sympy->JAX conversion;
jaxtyping
-
Writing Python like it's Rust
Try using [jaxtyping](https://github.com/google/jaxtyping).
It also supports numpy/pytorch/etc.
-
Writing Python like it’s Rust
Since you mention ML use-cases, you might like jaxtyping.
-
Scientific computing in JAX
jaxtyping: rich shape & dtype annotations for arrays and tensors (also supports PyTorch/TensorFlow/NumPy);
-
[D] Have their been any attempts to create a programming language specifically for machine learning?
Heads-up that my newer jaxtyping project now exists.
-
Returning to snake's nest after a long journey, any major advances in python for science ?
As other folks have commented, type hints are now a big deal. For static typing the best checker is pyright. For runtime checking there is typeguard and beartype. These can be integrated with array libraries through jaxtyping. (Which also works for PyTorch/numpy/etc., despite the name.)
- Type annotations and runtime checking for shape and dtype
What are some alternatives?
diffrax - Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
torchtyping - Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
sympytorch - Turning SymPy expressions into PyTorch modules.
MindsDB - The platform for customizing AI from enterprise data
plum - Multiple dispatch in Python
madtypes - Python Type that raise TypeError at runtime
pytype - A static type analyzer for Python code
cattrs - Composable custom class converters for attrs, dataclasses and friends.
hasktorch - Tensors and neural networks in Haskell
tiny-cuda-nn - Lightning fast C++/CUDA neural network framework
esm - Evolutionary Scale Modeling (esm): Pretrained language models for proteins