Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
☆37Jan 8, 2026Updated 5 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:
- Official implementation of Categorical Flow Maps on text.☆59Feb 16, 2026Updated 4 months ago
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 4 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 6 months ago
- Fast Discounted Cumulative Sums in PyTorch☆98Aug 28, 2021Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- DynGFN: Bayesian Dynamic Causal Discovery using Generative Flow Networks☆60Sep 17, 2025Updated 9 months ago
- Model predictive path integral control in jax☆17Feb 8, 2026Updated 4 months ago
- A Python package for linear subspace identification, nonlinear system identification, and nonlinear regression using Jax☆64May 18, 2026Updated last month
- Benchmark problems for decision-focused learning☆17Updated this week
- A modular implementation of PPO, and soon hopefully other algorithms.☆26Jan 16, 2024Updated 2 years ago
- EDIS: Energy-guided DIffusion Sampling☆19Aug 10, 2024Updated last year
- Visualize linear programming at https://lpviz.net☆41Updated this week
- ☆25Oct 9, 2022Updated 3 years ago
- Generative Flow Networks - GFlowNet☆337Updated this week
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- The online html book for the course modern control paradigms☆10Nov 7, 2025Updated 7 months ago
- Unofficial Julia package for Desmos☆24Jun 8, 2026Updated last week
- Learning to optimize (L2O) package that provides basic functionalities to help fit proxy models for optimization.☆15Apr 1, 2025Updated last year
- Standardised ROS messages for encoding whole-body state information☆14May 26, 2026Updated 3 weeks ago
- cuNumeric.jl wraps the cuPyNumeric C++ API providing a simple array programming interface that executes code on distributed clusters.☆20Jun 10, 2026Updated last week
- DAC: Diffusion Actor-Critic: Formulating Constrained Policy Iteration as Diffusion Noise Regression for Offline Reinforcement Learning.☆31Jun 3, 2024Updated 2 years ago
- [CVPR 2026 Highlight] XL-VLA: Cross-Hand Latent Representation for Vision-Language-Action Models☆97Apr 15, 2026Updated 2 months 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
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Implements vanilla PDHG algorithm☆10Mar 13, 2023Updated 3 years ago
- Docker container for ocropus3 OCR system☆12Aug 19, 2018Updated 7 years ago
- ☆18Sep 7, 2023Updated 2 years ago
- Decoupled Neural Interfaces Using Synthetic Gradients - under develeopment☆11Jun 27, 2025Updated 11 months ago
- Cloudflare workers friendly client for connecting to Apple's Push Notification Service using the new HTTP/2 protocol with JSON web tokens☆22Jan 4, 2025Updated last year
- Simple JAX Graphics Library.☆38Nov 3, 2024Updated last year
- Code of the paper "Universal Morphology Control via Contextual Modulation" at ICML 2023☆15Aug 3, 2023Updated 2 years ago
- PyPARC - A Python Package for Piecewise Affine Regression and Classification☆12Jul 4, 2023Updated 2 years ago
- ☆10Dec 21, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆15Jan 12, 2025Updated last year
- Log to W&B from Julia☆12Jun 13, 2022Updated 4 years ago
- [NeurIPS 2025 Spotlight] E2Former: An Efficient and Equivariant Transformer with Linear-Scaling Tensor Products☆29Feb 16, 2026Updated 4 months ago
- 320+ color themes for Hyper.☆18Jan 15, 2018Updated 8 years ago
- Tools for JAX☆50Updated this week
- Gaussian-Processes Surrogate Optimisation in python☆18May 18, 2021Updated 5 years ago
- Code for "Data-Efficient Reinforcement Learning with Self-Predictive Representations"☆166Dec 21, 2021Updated 4 years ago