Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
PyEPO: A PyTorch/JAX-based End-to-End Predict-then-Optimize Tool 2.2.5 documentation
PyEPO: A PyTorch/JAX-based End-to-End Predict-then-Optimize Tool 2.2.5 documentation

Basics

  • Introduction
  • Installation

Getting Started

  • Workflow
  • Optimization Models
  • Data and Datasets
  • Training Methods
  • Evaluation

Frontends and Backends

  • PyTorch Frontend
  • JAX Frontend
  • Solver Backends

Advanced Topics

  • Two-stage Baseline
  • Solution Pool
  • CaVE
  • kNN Robust Losses

Notebooks

  • Notebooks

API Reference

  • API Reference

Citation

  • Citation and References
  • API Reference
    • pyepo
      • pyepo.EPO
      • pyepo.data
        • pyepo.data.dataset
        • pyepo.data.knapsack
        • pyepo.data.portfolio
        • pyepo.data.shortestpath
        • pyepo.data.tsp
      • pyepo.dsl
        • pyepo.dsl.compiled
        • pyepo.dsl.expression
        • pyepo.dsl.objective
        • pyepo.dsl.problem
      • pyepo.func
        • pyepo.func.abcmodule
        • pyepo.func.blackbox
        • pyepo.func.cave
        • pyepo.func.contrastive
        • pyepo.func.jax
          • pyepo.func.jax.abcmodule
          • pyepo.func.jax.blackbox
          • pyepo.func.jax.cave
          • pyepo.func.jax.contrastive
          • pyepo.func.jax.perturbed
          • pyepo.func.jax.rank
          • pyepo.func.jax.regularized
          • pyepo.func.jax.surrogate
          • pyepo.func.jax.utils
        • pyepo.func.perturbed
        • pyepo.func.rank
        • pyepo.func.regularized
        • pyepo.func.runtime
        • pyepo.func.surrogate
        • pyepo.func.utils
      • pyepo.metric
        • pyepo.metric.metrics
        • pyepo.metric.mse
        • pyepo.metric.regret
        • pyepo.metric.unambregret
      • pyepo.model
        • pyepo.model.bases
        • pyepo.model.copt
          • pyepo.model.copt.compile
          • pyepo.model.copt.coptmodel
          • pyepo.model.copt.knapsack
          • pyepo.model.copt.portfolio
          • pyepo.model.copt.shortestpath
          • pyepo.model.copt.tsp
          • pyepo.model.copt.vrp
        • pyepo.model.grb
          • pyepo.model.grb.compile
          • pyepo.model.grb.grbmodel
          • pyepo.model.grb.knapsack
          • pyepo.model.grb.portfolio
          • pyepo.model.grb.shortestpath
          • pyepo.model.grb.tsp
          • pyepo.model.grb.vrp
        • pyepo.model.mpax
          • pyepo.model.mpax.compile
          • pyepo.model.mpax.knapsack
          • pyepo.model.mpax.mpaxmodel
          • pyepo.model.mpax.shortestpath
        • pyepo.model.omo
          • pyepo.model.omo.compile
          • pyepo.model.omo.knapsack
          • pyepo.model.omo.omomodel
          • pyepo.model.omo.portfolio
          • pyepo.model.omo.shortestpath
          • pyepo.model.omo.tsp
          • pyepo.model.omo.vrp
        • pyepo.model.opt
        • pyepo.model.ort
          • pyepo.model.ort.compile
          • pyepo.model.ort.knapsack
          • pyepo.model.ort.ortcpmodel
          • pyepo.model.ort.ortmodel
          • pyepo.model.ort.shortestpath
        • pyepo.model.predefined
        • pyepo.model.utils
      • pyepo.twostage
        • pyepo.twostage.autosklearnpred
        • pyepo.twostage.sklearnpred
      • pyepo.utils
Back to top
View this page

pyepo.func.jax¶

JAX autograd function for end-to-end training

Submodules¶

  • pyepo.func.jax.abcmodule
  • pyepo.func.jax.blackbox
  • pyepo.func.jax.cave
  • pyepo.func.jax.contrastive
  • pyepo.func.jax.perturbed
  • pyepo.func.jax.rank
  • pyepo.func.jax.regularized
  • pyepo.func.jax.surrogate
  • pyepo.func.jax.utils
Next
pyepo.func.jax.abcmodule
Previous
pyepo.func.contrastive
Copyright © 2021, Bo Tang
Made with Sphinx and @pradyunsg's Furo
On this page
  • pyepo.func.jax
    • Submodules