Bihaqo / tf_einsum_optLinks
Optimize the order of execution for tf.einsum
☆13Updated 8 years ago
Alternatives and similar repositories for tf_einsum_opt
Users that are interested in tf_einsum_opt are comparing it to the libraries listed below
Sorting:
- Implementation of deep implicit attention in PyTorch☆65Updated 4 years ago
- Unitary Evolution Recurrent Neural Network☆25Updated 3 years ago
- ☆18Updated 3 years ago
- CUDA-accelerated minimum spanning tree algorithm -- data parallel Boruvka's algorithm☆20Updated 9 years ago
- A smoother activation function (undergrad code)☆115Updated 5 years ago
- [ICML'21 Oral] Improving Lossless Compression Rates via Monte Carlo Bits-Back Coding☆14Updated 4 years ago
- SVG/Javascript-based library for creating presentations/figures☆60Updated 3 years ago
- Parallel implementation of Stochastic Gradient Descent using SciKit-Learn library in Python.☆26Updated 8 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆85Updated 4 years ago
- Official code for ICML 2020 paper "Variational Bayesian Quantization"☆24Updated 3 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- Implementation of ADMM-based sparse CNN architecture.☆12Updated 8 years ago
- ZORB: A Derivative-Free Backpropagation Algorithm for Neural Networks☆22Updated 5 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆38Updated last year
- Code publication to the paper "Normalized Attention Without Probability Cage"☆17Updated 4 years ago
- Notebooks for IPAM Tutorial, March 15 2019☆24Updated 6 years ago
- Efficient Householder Transformation in PyTorch☆69Updated 4 years ago
- Deterministic Decoding for Discrete Data in Variational Autoencoders☆24Updated 5 years ago
- [ICLR 2019] Learning Representations of Sets through Optimized Permutations☆36Updated 6 years ago
- Structured matrices for compressing neural networks☆67Updated 2 years ago
- Sample pytorch implementation of Covariant Compositional Networks☆13Updated 7 years ago
- Word2ket: Space-efficient Word Embeddings inspired by Quantum Entanglement☆52Updated 5 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Updated 3 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- Simple gradient checkpointing for eager mode execution☆46Updated 5 years ago
- ☆34Updated 6 years ago
- Interpretable Models for NLP using PyTorch☆18Updated 7 years ago
- Deep Boltzmann Machines in R^N dimensions☆11Updated 11 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆14Updated 6 years ago