hengyuan-hu / jax-vs-pytorch
☆10Updated last year
Related projects ⓘ
Alternatives and complementary repositories for jax-vs-pytorch
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆43Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆26Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆42Updated 3 weeks ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated last year
- ☆14Updated 2 years ago
- Offline RL experiments☆14Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 3 months ago
- An adaptive training algorithm for residual network☆14Updated 4 years ago
- Cross-Domain Imitation Learning via Optimal Transport☆23Updated 2 years ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated last year
- ☆40Updated last year
- Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…☆9Updated 3 years ago
- Implementation of ICML 2023 paper: Future-conditioned Unsupervised Pretraining for Decision Transformer☆27Updated last year
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆34Updated last year
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆26Updated 2 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Updated 3 years ago
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- [NeurIPS 2022] Compositional Generalization in Unsupervised Compositional Representation Learning: A Study on Disentanglement and Emergen…☆13Updated 2 years ago
- "CoPhy: Counterfactual Learning of Physical Dynamics", F. Baradel, N. Neverova, J. Mille, G. Mori, C. Wolf, ICLR'2020☆33Updated 4 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- Code for ICML2021 paper 'Commutative Lie Group VAE for Disentanglement Learning'.☆23Updated 2 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated last year
- ☆28Updated 8 months ago
- Structured Object-Aware Physics Prediction for Video Modeling and Planning☆32Updated 4 years ago
- An Elegant Library for Bayesian Deep Learning in PyTorch☆22Updated last year
- ☆23Updated 3 years ago
- Code for Abstract-to-Executable Trajectory Translation for One Shot Task Generalization (ICML 2023)☆22Updated last year
- params_proto, a collection of decorators that makes shell argument passing declarative☆13Updated 7 months ago
- ☆29Updated 2 years ago