hengyuan-hu / jax-vs-pytorchLinks
☆12Updated 3 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☆27Updated 3 years ago
- ☆15Updated 2 years ago
- The code for the paper "A Bayesian Approach to Online Planning" published in ICML 2024.☆11Updated 11 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated 2 years ago
- An adaptive training algorithm for residual network☆15Updated 4 years ago
- ☆10Updated 2 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- ☆24Updated last year
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆42Updated 7 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- implementation of dualformer☆17Updated 3 months ago
- [ICLR'25] "Understanding Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing" by Peihao Wang, Ruisi Cai, Yue…☆11Updated 2 months ago
- ☆12Updated 2 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆17Updated last year
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- Code for LaMPP: Language Models as Probabilistic Priors for Perception and Action☆37Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 10 months ago
- ☆17Updated last year
- Code for ICML2021 paper 'Commutative Lie Group VAE for Disentanglement Learning'.☆22Updated 2 years ago
- Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…☆9Updated 4 months ago
- This is the pytorch implementation of the UAI2023 paper "A Trajectory is Worth Three Sentences: Multimodal Transformer for Offline Reinf…☆11Updated last year
- ☆34Updated 2 months ago
- Implementation of ICML 2023 paper: Future-conditioned Unsupervised Pretraining for Decision Transformer☆27Updated last year
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated last year
- ☆15Updated 2 years ago
- Generalised UDRL☆37Updated 3 years ago
- flexible meta-learning in jax☆14Updated last year
- Official Release of NeurIPS 2020 Spotlight paper "Generative Neurosymbolic Machines"☆35Updated last year