Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
☆37Jan 8, 2026Updated 2 months ago
Alternatives and similar repositories for torch_jax_interop
Users that are interested in torch_jax_interop are comparing it to the libraries listed below
Sorting:
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 3 months ago
- ☆21Jan 24, 2025Updated last year
- ☆19May 18, 2024Updated last year
- ☆19May 11, 2023Updated 2 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆26Feb 2, 2025Updated last year
- Fast Discounted Cumulative Sums in PyTorch☆96Aug 28, 2021Updated 4 years ago
- GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…☆86Feb 15, 2023Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Feb 15, 2026Updated 3 weeks ago
- ☆25Oct 9, 2022Updated 3 years ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆29Oct 12, 2023Updated 2 years ago
- Simple JAX Graphics Library.☆36Nov 3, 2024Updated last year
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆79Nov 14, 2023Updated 2 years ago
- W&B Hyperparameter Sweep Engine. File sweeps related issues at the W&B client: https://github.com/wandb/client☆40Jan 14, 2025Updated last year
- ☆41Apr 27, 2022Updated 3 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago
- About The dataset was recorded on the Husky robotics platform on the university campus and consists of 5 tracks recorded at different tim…☆11Mar 25, 2025Updated 11 months ago
- Creates Random Coding Sequences with specified GC content and Amino Acid usage☆10Jun 21, 2022Updated 3 years ago
- Some kind of TidalCycles implementation for SuperCollider☆14May 29, 2020Updated 5 years ago
- Tool to convert '.com' Gaussian files into files supported by 3D rendering programs, such as Blender, Maya, and others.☆14Updated this week
- Partially Observable Multi-Agent RL with Transformers☆17Feb 13, 2026Updated 3 weeks ago
- A data processing module implemented with numpy☆10Aug 16, 2022Updated 3 years ago
- ☆15Jul 27, 2023Updated 2 years ago
- A C++ template library for polynomials algebra over discrete integral domains☆11May 9, 2025Updated 10 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆273Sep 22, 2025Updated 5 months ago
- Multimodal emotion recognition is a challenging task because emotions can be expressed through various modalities. It can be applied in v…☆10Jun 27, 2019Updated 6 years ago
- ☆11Feb 2, 2024Updated 2 years ago
- A Type System Attempt for Wolfram Language☆13Feb 28, 2021Updated 5 years ago
- DSA Dart Broker☆10Oct 22, 2020Updated 5 years ago
- Exploratory search engine based on hierarchical topic models from BigARTM☆13Mar 8, 2022Updated 4 years ago
- DARPA Cyber Grand Challenge Linux source code☆17Jul 9, 2015Updated 10 years ago
- Zig library for the Ollama API.☆13Oct 17, 2025Updated 4 months ago
- This repo contains all the code related to the interactive online Bloch sphere simulator☆18Updated this week
- ☆20Dec 1, 2025Updated 3 months ago
- A python wrapper for ScriptHookV☆11Jun 26, 2018Updated 7 years ago
- Some commonly used functions and modules☆10Jan 15, 2024Updated 2 years ago
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆17Jul 11, 2023Updated 2 years ago
- A tool for migrating settings that PlasmaConfigSaver is not taking care of☆12Dec 16, 2019Updated 6 years ago
- ☆10Aug 26, 2022Updated 3 years ago
- Functional Iterator for python☆11May 17, 2021Updated 4 years ago