Bihaqo / tf_einsum_opt
Optimize the order of execution for tf.einsum
☆14Updated 7 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
- Variational Walkback, NIPS'17☆28Updated 7 years ago
- ☆18Updated 3 years ago
- Rotational Unit of Memory☆34Updated 6 years ago
- ☆12Updated 6 years ago
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 5 years ago
- ☆30Updated 3 years ago
- Code for the paper "Learning sparse transformations through backpropagation"☆43Updated 5 years ago
- Run Pytorch graphs inside Theano graph (and pytorch wrapper for AIS for generative models).☆18Updated 7 years ago
- Successfully training approximations to full-rank matrices for efficiency in deep learning.☆17Updated 4 years ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions☆22Updated 5 years ago
- Open source implementation of SeaRNN (ICLR 2018, https://openreview.net/forum?id=HkUR_y-RZ)☆48Updated 6 years ago
- An implementation of various tensor-based decomposition for NN & RNN parameters☆18Updated 6 years ago
- Code for the 2-simplicial Transformer paper☆18Updated 5 years ago
- Reference implementation of algorithms for reinforcement learning and Markov decision processes.☆11Updated 4 years ago
- Efficient Unitary Neural Network(EUNN) implementation in Tensorflow☆73Updated 5 years ago
- Code repo for "Transformer on a Diet" paper☆31Updated 4 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆47Updated 5 years ago
- Matlab code implementing Minimum Probability Flow Learning.☆68Updated 10 years ago
- Cleaned research code for predicting table structure from an image.☆9Updated 8 years ago
- Variational autoencoder in Theano