A language for mental models
Multiple dispatch over abstract array types in JAX.
What did I do on February 14th 2007? Visualize your (digital) life in Org-mode
JAX-based neural network library
Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
Train very large language models in Jax.
Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
A JAX research toolkit for building, editing, and visualizing neural networks.
Concise deep learning for JAX
Explore large language models in 512MB of RAM
A memoize library which can be used standalone, or plugged into key/value stores such as redis. A...
Turn SymPy expressions into trainable JAX expressions.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Extending JAX with custom C++ and CUDA code
CLI & Python API to easily summarize text-based files with transformers
High-Performance Klong array language in Python.