arogozhnikov / eindexLinks
Multidimensional indexing for tensors
β139Updated 2 years ago
Alternatives and similar repositories for eindex
Users that are interested in eindex are comparing it to the libraries listed below
Sorting:
- β60Updated 3 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with π€`safetensors`β47Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β188Updated this week
- TorchFix - a linter for PyTorch-using code with autofix supportβ152Updated 3 months ago
- LoRA for arbitrary JAX models and functionsβ143Updated last year
- Experiment of using Tangent to autodiff tritonβ81Updated last year
- A functional training loops library for JAXβ88Updated last year
- This is a port of Mistral-7B model in JAXβ32Updated last year
- Named tensors with first-class dimensions for PyTorchβ332Updated 2 years ago
- Running Jax in PyTorch Lightningβ114Updated 11 months ago
- Automatically take good care of your preemptible TPUsβ37Updated 2 years ago
- β44Updated last month
- β62Updated last year
- OpTree: Optimized PyTree Utilitiesβ202Updated this week
- β91Updated last year
- A library for unit scaling in PyTorchβ132Updated 5 months ago
- A Pytree Module system for Deep Learning in JAXβ214Updated 2 years ago
- β118Updated this week
- Unofficial JAX implementations of deep learning research papersβ159Updated 3 years ago
- JMP is a Mixed Precision library for JAX.β211Updated 10 months ago
- Implementation of GateLoop Transformer in Pytorch and Jaxβ91Updated last year
- Mobile Viewer for W&B, built on top of Flutter.β39Updated last year
- If it quacks like a tensor...β59Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAXβ181Updated 7 months ago
- Implementation of Flash Attention in Jaxβ222Updated last year
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)β52Updated last year
- A Python package of computer vision models for the Equinox ecosystem.β110Updated last year
- β107Updated last year
- Neural Networks for JAXβ84Updated last year
- Universal Notation for Tensor Operations in Python.β452Updated 8 months ago