Welcome to OJAX’s documentation!

+-------------------------------------+
|       __              __      __    |
|      /_/\            / /\    /_/\   |
|     /_/  \          / /  \  /_/ /   |
|    /_/ /\ \        / / /\ \/_/ /    |
|   /_/ /\_\ \      / / /\_\ \/ /     |
|  /_/ /  \_\ \    / / /__\_\  /      |
|  \_\ \  /_/ /   / / ______   \      |
|   \_\ \/_/ /   / / /   /_/ /\ \     |
|    \_\ \/_/___/ / /   /_/ /\_\ \    |
|     \_\/_______/ /   /_/ /  \_\ \   |
|      \_\_\_\_\_\/    \_\/    \_\/   |
|                                     |
+-------------------------------------+

• Introduction
  • What is OJAX
  • Why OJAX
  • How to code with OJAX
• Installing OJAX
  • Dependencies
  • Installing OJAX
• OJAX Quickstart
  • Declaring annotated fields
  • Field types for OTree
  • The __init__ method
  • Updating the fields
  • Adding JAX methods
  • General coding tips
• Example: NN layer with OJAX
• API Reference
  • ojax package
• Change log
  • ojax 3.1.0
  • ojax 3.0.0
  • ojax 2.0.2
  • ojax 2.0.1
  • ojax 2.0.0
  • ojax 1.0.0

Indices and tables

• Index

• Module Index

• Search Page