{"id":26046698,"url":"https://github.com/ickc/python-autojax","last_synced_at":"2026-04-22T13:31:49.049Z","repository":{"id":273921063,"uuid":"921273969","full_name":"ickc/python-autojax","owner":"ickc","description":"DiRAC: revealing the nature of dark matter with the James Webb space telescope and JAX","archived":false,"fork":false,"pushed_at":"2025-09-17T14:08:24.000Z","size":457,"stargazers_count":0,"open_issues_count":0,"forks_count":0,"subscribers_count":1,"default_branch":"main","last_synced_at":"2026-04-07T06:38:44.237Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":"https://ickc.github.io/python-autojax/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"bsd-3-clause","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/ickc.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null}},"created_at":"2025-01-23T16:54:17.000Z","updated_at":"2025-09-17T14:08:27.000Z","dependencies_parsed_at":null,"dependency_job_id":"3110f58c-ca6b-4db1-8b92-305f7f4be242","html_url":"https://github.com/ickc/python-autojax","commit_stats":null,"previous_names":["ickc/python-autojax"],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/ickc/python-autojax","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ickc%2Fpython-autojax","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ickc%2Fpython-autojax/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ickc%2Fpython-autojax/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ickc%2Fpython-autojax/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/ickc","download_url":"https://codeload.github.com/ickc/python-autojax/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ickc%2Fpython-autojax/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":32139145,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-04-22T13:27:12.868Z","status":"ssl_error","status_checked_at":"2026-04-22T13:26:44.791Z","response_time":58,"last_error":"SSL_read: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2025-03-07T21:12:28.722Z","updated_at":"2026-04-22T13:31:49.009Z","avatar_url":"https://github.com/ickc.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# DiRAC: revealing the nature of dark matter with the James Webb space telescope and JAX\n\n![GitHub Actions](https://github.com/ickc/python-autojax/workflows/Unit%20tests/badge.svg)\n[![Documentation Status](https://github.com/ickc/python-autojax/workflows/GitHub%20Pages/badge.svg)](https://ickc.github.io/python-autojax)\n\n[![GitHub Releases](https://img.shields.io/github/tag/ickc/python-autojax.svg?label=github+release)](https://github.com/ickc/python-autojax/releases)\n\u003c!-- [![PyPI Package latest release](https://img.shields.io/pypi/v/autojax.svg)](https://pypi.org/project/autojax)\n[![Supported versions](https://img.shields.io/pypi/pyversions/autojax.svg)](https://pypi.org/project/autojax)\n[![Supported implementations](https://img.shields.io/pypi/implementation/autojax.svg)](https://pypi.org/project/autojax)\n[![PyPI Wheel](https://img.shields.io/pypi/wheel/autojax.svg)](https://pypi.org/project/autojax)\n[![Development Status](https://img.shields.io/pypi/status/autojax.svg)](https://pypi.python.org/pypi/autojax/)\n[![Downloads](https://img.shields.io/pypi/dm/autojax.svg)](https://pypi.python.org/pypi/autojax/)\n![License](https://img.shields.io/pypi/l/autojax.svg) --\u003e\n\n# Introduction\n\nThis is a small self-contained repo for the 3 months project \"DiRAC: revealing the nature of dark matter with the James Webb space telescope and JAX\". It is organized in 3 modules. `original` is copied from [Jammy2211/dirac_rse_interferometer](https://github.com/Jammy2211/dirac_rse_interferometer) (which itself is copied from various other repos such as AutoGalaxy and AutoLens), and then they are ported to JAX in the `jax` module. While the `original` is already implemented in Numba, a `numba` module is also provided here, mainly as a starting point to port from `original` to something more vectorized first. Often time the `jax` implementation is the same as the `numba` implementation here, or a close variant of it due to differences between Numba and JAX.\n\nAs part of the goal to port to JAX implementation is to speed up, benchmark is provided to compare the 3 implementations. See instructions below to see how to run it.\n\nFor a logbook style repo that contains every notes about the project and how to run the AutoGalaxy family of softwares, see [ickc/log-PyAutoLens](https://github.com/ickc/log-PyAutoLens). There's some scripts under `external/` here that can only be run using the environment documented there, as it uses some external (to us) dependencies recorded as git submodules over there.\n\n# Installing the project\n\n## pip\n\n```sh\npip install -e .[tests]\n```\n\n## conda/mamba\n\n```sh\nconda env create -f environment.yml\nconda activate autojax\n# update using\nconda env update --name autojax --file environment.yml --prune\n```\n\n## pixi\n\n```sh\npixi install\n# prepend everything you run by pixi run, such as\npixi run pytest\n```\n\n:::{Note}\nFrom this point on, pixi is assumed and every command is prepended with `pixi run`. For other ways of loading environment, simply load your environment as usual and disregard this prefix.\n:::\n\n# Running unit tests and benchmarks\n\n```sh\npytest\n# or if you use pixi\npixi run pytest\n```\n\nThis should runs the tests and also give you benchmark information comparing different implementations.\n\nBy default, it will run the benchmark as well. If you want to run unit tests only,\n\n```sh\npytest --benchmark-disable\n# or if you use pixi\npixi run pytest --benchmark-disable\n```\n\nThere are 2 benchmarks designed to test various different implementations to calculate `w_tilde` and `curvature_matrix`, `TestWTilde` and `TestCurvatureMatrix` respectively.\nYou can filter them like this,\n\n```sh\npytest -k 'TestWTilde or TestCurvatureMatrix'\n# or if you use pixi\npixi run pytest -k 'TestWTilde or TestCurvatureMatrix'\n```\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fickc%2Fpython-autojax","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fickc%2Fpython-autojax","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fickc%2Fpython-autojax/lists"}