{"categories":[{"label":"Artificial Intelligence","url":"https://skillfed.io/packages/category/scientific-engineering-artificial-intelligence/4"},{"label":"Information Analysis","url":"https://skillfed.io/packages/category/scientific-engineering-information-analysis"},{"label":"Mathematics","url":"https://skillfed.io/packages/category/scientific-engineering-mathematics/2"}],"enrichment":{"capability":"Equinox provides neural network and model building on top of JAX with PyTorch-like syntax, plus PyTree manipulation, filtered transformations, and runtime error handling\u2014all while remaining fully compatible with core JAX operations.","skillfed_tags":["jax-ecosystem","neural-networks","functional-programming"],"use_cases":["Building and training neural networks in JAX with PyTorch-familiar class syntax while preserving JAX's functional composition.","Defining custom models as PyTrees that can be passed directly through jit-compiled and grad-transformed functions.","Manipulating model parameters and structure using Equinox's PyTree utilities without manual pytree registration.","Prototyping deep learning research where you need advanced JAX features like vmap and grad but want simpler model definition syntax.","Migrating from Flax or Haiku to JAX while retaining model-building ergonomics and gaining access to lower-level JAX control."],"what_it_does":"Equinox is a JAX library that fills gaps in core JAX for machine learning by providing neural network and model abstractions with familiar PyTorch-like syntax. Models are defined as PyTrees\u2014JAX's native data structure\u2014so they integrate seamlessly with JAX transformations like jit, grad, and vmap without special handling. The library includes utilities for PyTree manipulation, filtered APIs for transformations, and runtime error support.\n\nUnlike frameworks, Equinox does not enforce a specific training loop or impose constraints on how you use JAX. Everything you write remains compatible with the broader JAX ecosystem. It depends on jax, jaxtyping, typing-extensions, and wadler-lindig, all of which are lightweight. The package is actively maintained and positioned for developers who want neural network convenience without sacrificing JAX's composability and functional programming model.","worth_installing":"Yes. Equinox is actively maintained, has no known vulnerabilities, installs with low friction, and offers a permissive Apache 2.0 license. It fills a genuine gap for developers who want neural network abstractions in JAX without framework overhead. The main caveat is the Alpha development status and the requirement for Python 3.10+; if you need production stability or support for older Python versions, verify API stability first."},"id":"equinox","links":{"html":"https://skillfed.io/packages/equinox","md":"https://skillfed.io/packages/equinox.md","pypi":"https://pypi.org/project/equinox/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2026-05-05","license_spdx":null,"license_treatment":"permissive","name":"equinox","python_support":"supports_current","summary":"Elegant easy-to-use neural networks in JAX."},"popularity":{"monthly_downloads":1232737,"position":4183,"tier":"top_5000"},"security":{"n_vulnerabilities":0},"version":"0.13.8"}