Similar repositories to awf/awf-jaxutils:

• awf/awf-jaxutils github similar
• awf/awfutils github similar
• openxla/community github similar
• cgarciae/treeo github similar
• ucl-bug/jaxdf github similar
• tsattler/meshloc_release github similar
• gehring/fax github similar
• paschalidoud/hierarchical_primitives github similar
• nmwsharp/neural-implicit-queries github similar
• brentyi/jaxfg github similar
• google/tree-math github similar
• facebookresearch/neural-light-fields github similar
• CarloLucibello/GraphNeuralNetworks.jl github similar
• brentyi/jaxlie github similar
• awf/functional-transformer github similar
• paschalidoud/neural_parts github similar
• Boese0601/RC-MVSNet github similar
• tovacinni/sdf-explorer github similar
• ericjang/maml-jax github similar
• google/jaxtyping github similar
• ott-jax/ott github similar
• cgarciae/treex github similar
• dfm/extending-jax github similar
• jacobjinkelly/easy-neural-ode github similar
• srush/streambook github similar
• didriknielsen/survae_flows github similar
• wesselb/plum github similar
• RobertTLange/gymnax github similar
• MilesCranmer/lagrangian_nns github similar
• deepmind/distrax github similar
• google/jax-cfd github similar
• JuliaDiff/Diffractor.jl github similar
• lilipads/gradient_descent_viz github similar
• google/jaxopt github similar
• deepmind/chex github similar
• pytorch/torchdynamo github similar
• zju3dv/NeuralRecon-W github similar
• google/evojax github similar
• threedle/text2mesh github similar
• patrick-kidger/diffrax github similar