microJAX

microJAX is a differentiable microlensing library built with JAX. It provides point-source calculations for one to three lenses, finite-source calculations for binary and triple lenses, and trajectory utilities.

The finite-source API includes a GPU-oriented accelerator backend and a separate CPU backend for binary lenses. Both support uniform and linear limb-darkened circular sources. Use Getting Started to install the package and verify the JAX environment, then continue with Usage Guide for public API examples.

Guides

• Getting Started
  • Prerequisites
  • Installation
  • Verify the environment
  • Up next
• Usage Guide
  • Common setup
  • Point-source magnification
  • Finite-source point lens
  • Finite-source binary lenses
  • Triple lenses
  • Configuration
  • Differentiation
  • Failure behavior
• CPU Binary-Lens Backend
  • Basic use
  • Numerical contract
  • Forward-mode differentiation
  • Timing
  • Examples
• Accelerator Performance Tuning
  • Static scheduler settings
  • Benchmark the analysis workload
  • Memory and accuracy
• Solver Caveats
  • Backend scope is explicit
  • Finite values are not accuracy certificates
  • Rejected configurations return NaN
  • The numerical graph is piecewise
  • Use x64 mode and forward differentiation
  • Configuration changes compile separately
  • The paper version is a different solver line
• Citing and Reproducibility
  • Methods paper and software archive
  • BibTeX
  • Reproducible numerical results

API Reference

• API Reference

Choosing a guide

• CPU Binary-Lens Backend defines the binary-lens CPU interface, supported differentiation, and numerical contract.

• Accelerator Performance Tuning explains accelerator scheduler settings and sound timing practice.

• Solver Caveats collects microJAX-specific limitations that can affect a scientific result.

• Citing and Reproducibility gives citations and the reproducibility information to report.

The methods-paper implementation is archived as v0.1.1. The 0.2 series is a substantial solver redesign, so reproducible work should record the exact microJAX release or Git commit.

Indices and tables

• Index

• Module Index

• Search Page