Skip to main content

Ctrl+K

chemtrain

Documentation Overview

Algorithms

Differentiable Trajectory Reweighting (DiffTRe)
Force Matching
Relative Entropy Minimization
Prior Simulation

Examples

DiffTRe: CG Water on Structural Data
Alanine Dipeptide in Implicit Water
Bottom-Up and Top-Down Training of Atomistic Titanium

Chemtrain

compose
- compose.mace_jax
- compose.utils
data
- data.preprocessing
- data.data_loaders
deploy
- deploy.exporter
- deploy.graphs
learn
quantity
ensemble
trainers
typing

chemtrain-deploy

Installation
Getting Started
API: Connector
Developer Information

Jax, M.D. Extensions

model
custom_electrostatics
custom_energy
- jax_md_mod.custom_energy.lorentz_berthelot
custom_interpolate
custom_partition
custom_quantity
custom_space
io

.rst

compose.mace_jax

Contents

Species Mappings
Load Foundational Models

`compose.mace_jax`#

Species Mappings#

Load Foundational Models#

previous

compose

next

compose.utils

Contents

Species Mappings
Load Foundational Models

By Multiscale Modeling of Fluid Materials

© Copyright 2026, Multiscale Modeling of Fluid Materials, TU Munich.