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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Tools to connect to and interact with the Mila cluster☆79Updated this week
- ☆21Jan 24, 2025Updated last year
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 3 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆26Feb 2, 2025Updated last year
- GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…☆86Feb 15, 2023Updated 3 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 4 months ago
- Code for "What really matters in matrix-whitening optimizers?"☆23Oct 31, 2025Updated 4 months ago
- ☆19May 18, 2024Updated last year
- ☆19May 11, 2023Updated 2 years ago
- Fast Discounted Cumulative Sums in PyTorch☆97Aug 28, 2021Updated 4 years ago
- DynGFN: Bayesian Dynamic Causal Discovery using Generative Flow Networks☆60Sep 17, 2025Updated 6 months ago
- A modular, easy to extend GFlowNet library☆305Updated this week
- ☆55Aug 9, 2024Updated last year
- Model predictive path integral control in jax☆17Feb 8, 2026Updated last month
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- ☆10Feb 17, 2026Updated last month
- A Python package for linear subspace identification, nonlinear system identification, and nonlinear regression using Jax☆59Updated this week
- Benchmark problems for decision-focused learning☆15Updated this week
- A modular implementation of PPO, and soon hopefully other algorithms.☆26Jan 16, 2024Updated 2 years ago
- Rethinking the Trust Region in LLM Reinforcement Learning☆52Mar 2, 2026Updated 3 weeks ago
- Research Project Template Repository☆44Jan 13, 2026Updated 2 months ago
- ☆41Apr 27, 2022Updated 3 years ago
- ☆13Apr 7, 2022Updated 3 years ago
- Visualize linear programming at https://lpviz.net☆33Jan 20, 2026Updated 2 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- ☆25Oct 9, 2022Updated 3 years ago
- World-Gymnast: Training Robots with Reinforcement Learning in a World Model☆30Feb 11, 2026Updated last month
- Generative Flow Networks - GFlowNet☆322Mar 22, 2026Updated last week
- Official implementation of Categorical Flow Maps on text.☆49Feb 16, 2026Updated last month
- An adaptive/incremental kNN and Range Search query implementation in Julia☆11May 6, 2023Updated 2 years ago
- [ARCHIVED] Development continues at https://github.com/Simple-Robotics/aligator. [Old description] A primal-dual augmented Lagrangian sol…☆68Sep 26, 2025Updated 6 months ago
- ☆22Nov 11, 2023Updated 2 years ago
- DAC: Diffusion Actor-Critic: Formulating Constrained Policy Iteration as Diffusion Noise Regression for Offline Reinforcement Learning.☆27Jun 3, 2024Updated last year
- Run PyTorch in JAX. 🤝☆314Oct 13, 2025Updated 5 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- ☆10Aug 26, 2022Updated 3 years ago
- cuNumeric.jl wraps the cuPyNumeric C++ API providing a simple array programming interface that executes code on distributed clusters.☆19Mar 3, 2026Updated 3 weeks ago
- Deep RL agents with PyTorch☆36Sep 25, 2021Updated 4 years ago
- High-level Python Particle Sequential Convex Programming Model Predictive Control (SCP PMPC) interface☆17Oct 30, 2023Updated 2 years ago
- ☆18Dec 6, 2023Updated 2 years ago
- Implements vanilla PDHG algorithm☆10Mar 13, 2023Updated 3 years ago
- ☆18Sep 7, 2023Updated 2 years ago