tensor_annotations
jaxtyping
tensor_annotations | jaxtyping | |
---|---|---|
2 | 7 | |
158 | 955 | |
-0.6% | 5.3% | |
5.8 | 8.3 | |
10 months ago | 19 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.
tensor_annotations
-
[D] Have their been any attempts to create a programming language specifically for machine learning?
Not really an answer to your question, but there are Python packages that try to solve the problem of tensor shapes that you mentioned, e.g. https://github.com/patrick-kidger/torchtyping or https://github.com/deepmind/tensor_annotations
-
Matrix Multiplication Inches Closer to Mythic Goal
I've explored this space quite a bit. In my view, static checking should be the goal.
https://github.com/deepmind/tensor_annotations and tsastanley seem to be the most far along. I've developed a mypy plugin that does similarly off of the "Named Tensor" dynamic feature (which isn't well supported yet), but haven't released it yet.
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?
dex-lang - Research language for array processing in the Haskell/ML family
torchtyping - Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
MindsDB - The platform for customizing AI from enterprise data
miniF2F - Formal to Formal Mathematics Benchmark
diffrax - Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
TablaM - The practical relational programing language for data-oriented applications
plum - Multiple dispatch in Python
tiny-cuda-nn - Lightning fast C++/CUDA neural network framework
madtypes - Python Type that raise TypeError at runtime
pytype - A static type analyzer for Python code