fferflo / einx
Universal Tensor Operations in Einstein-Inspired Notation for Python.
☆328Updated last month
Related projects ⓘ
Alternatives and complementary repositories for einx
- Run PyTorch in JAX. 🤝☆200Updated last year
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Scalable neural net training via automatic normalization in the modular norm.☆121Updated 3 months ago
- ☆303Updated this week
- For optimization algorithm research and development.☆449Updated this week
- Multidimensional indexing for tensors☆113Updated last year
- ☆197Updated 4 months ago
- Accelerated First Order Parallel Associative Scan☆163Updated 3 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆343Updated 3 weeks ago
- OpTree: Optimized PyTree Utilities☆152Updated this week
- A simple library for scaling up JAX programs☆127Updated 2 weeks ago
- 94% on CIFAR-10 in 2.6 seconds 💨 96% in 27 seconds☆177Updated last week
- ☆292Updated 4 months ago
- Annotated version of the Mamba paper☆457Updated 8 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆333Updated 3 weeks ago
- ☆128Updated this week
- Use Jax functions in Pytorch☆228Updated last year
- Efficient optimizers☆79Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆252Updated 5 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆172Updated 2 months ago
- Helpful tools and examples for working with flex-attention☆469Updated 3 weeks ago
- TensorDict is a pytorch dedicated tensor container.☆840Updated last week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,219Updated this week
- ☆139Updated 3 months ago
- Named Tensors for Legible Deep Learning in JAX☆153Updated this week
- CLU lets you write beautiful training loops in JAX.☆321Updated this week
- LoRA for arbitrary JAX models and functions☆132Updated 8 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆516Updated this week
- ☆207Updated 6 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆303Updated this week