kachayev / dataclasses-tensorLinks
Easily serialize dataclasses to and from tensors (PyTorch, NumPy)
☆18Updated 4 years ago
Alternatives and similar repositories for dataclasses-tensor
Users that are interested in dataclasses-tensor are comparing it to the libraries listed below
Sorting:
- A framework for implementing equivariant DL☆10Updated 4 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"☆13Updated 4 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆16Updated 4 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆49Updated 3 years ago
- PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆17Updated 3 years ago
- Official implementation of the paper "Interventions, Where and How? Experimental Design for Causal Models at Scale", NeurIPS 2022.☆20Updated 2 years ago
- High-performance tokenized language data-loader for Python C++ extension☆13Updated last year
- Visualize tensors in a plain Python REPL using Sparklines☆45Updated 4 years ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Updated 2 years ago
- ☆31Updated 4 months ago
- Large dataset storage format for Pytorch☆45Updated 4 years ago
- ☆37Updated 3 years ago
- ☆15Updated 3 years ago
- ☆30Updated 3 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- AdaCat☆49Updated 3 years ago
- Composable kernels for scikit-learn implemented in JAX.☆45Updated 5 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- A stateful pytree library for training neural networks.☆22Updated last month
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- Test pytorch code with minimal computational overhead☆26Updated 2 years ago
- ☆15Updated 4 years ago
- Quasi-Newton Algorithm for Stochastic Optimization☆10Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- NeurIPS 2019 Paper Implementation☆12Updated 2 years ago
- List of awesome JAX resources☆13Updated 2 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Ludwig benchmark☆19Updated 3 years ago