Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/brentyi/pips-jax
/ types & classes
Types & classes
8 in github.com/brentyi/pips-jax
⨍
Functions
27
◇
Types & classes
8
↓ 3 callers
Class
Pips
src/pips_jax/model.py:33
↓ 3 callers
Class
Timer
Timing helper. Usage: with Timer() as t: ... # Do something. print(t.elapsed) # seconds
benchmark.py:20
↓ 2 callers
Class
ResidualBlock
src/pips_jax/model.py:342
↓ 2 callers
Class
_TrackedParticles
Internal state tracked by our PIPs model. Each particle is represented by a feature and a coordinate.
src/pips_jax/model.py:25
↓ 1 callers
Class
BasicEncoder
Basic encoder from RAFT.
src/pips_jax/model.py:239
↓ 1 callers
Class
DeltaBlock
src/pips_jax/model.py:401
↓ 1 callers
Class
MLPMixer
src/pips_jax/model.py:429
Class
BenchmarkConfig
benchmark.py:40