{
  "id": "library:jax",
  "_kind": "Library",
  "_file": "domain/libraries/libraries-python.yaml",
  "_cluster": "domain",
  "attributes": {
    "displayName": "JAX",
    "belongsToLanguageId": "language:python",
    "description": "NumPy-compatible array library with grad/jit/vmap/pmap; backbone of\nFlax and many high-performance ML research codebases.\n"
  },
  "outgoingEdges": [
    {
      "from": "library:jax",
      "to": "language:python",
      "kind": "belongs_to_language"
    },
    {
      "from": "library:jax",
      "to": "skill-area:machine-learning",
      "kind": "library_used_by",
      "attributes": {}
    }
  ],
  "incomingEdges": [
    {
      "from": "library:pytorch",
      "to": "library:jax",
      "kind": "alternative_to"
    }
  ]
}