hengyuan-hu / jax-vs-pytorchLinks
☆13Updated 8 months ago
Alternatives and similar repositories for jax-vs-pytorch
Users that are interested in jax-vs-pytorch are comparing it to the libraries listed below
Sorting:
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆28Updated 4 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆46Updated 2 years ago
- An adaptive training algorithm for residual network☆17Updated 5 years ago
- ☆10Updated 2 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆25Updated 2 months ago
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated 4 months ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated 2 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Updated last year
- Cross-Domain Imitation Learning via Optimal Transport☆25Updated 3 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆62Updated 3 years ago
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆21Updated last year
- Official code for the paper "Attention as a Hypernetwork"☆44Updated last year
- ☆15Updated 3 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆43Updated 2 years ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆55Updated 10 months ago
- Generalised UDRL☆37Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆43Updated 11 months ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- ☆15Updated 3 years ago
- Variational Reinforcement Learning☆16Updated last year
- ☆13Updated 2 years ago
- ☆34Updated 11 months ago
- ☆12Updated last year
- Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers☆23Updated 7 months ago
- ☆34Updated 7 months ago
- Meta Optimal Transport☆104Updated 2 years ago
- Code for LaMPP: Language Models as Probabilistic Priors for Perception and Action☆37Updated 2 years ago