Skip to content

issues Search Results · repo:google/tree-math language:Python

Filter by

11 results
 (69 ms)

11 results

ingoogle/tree-math (press backspace or delete to remove)

google/tree-math
support static fields in treemath.struct

it would be nice to have some fields be pytree nodes and others not, which would make this a full replacement for flax.struct
  • GallagherCommaJack
  • 4
  • Opened 
    on Jul 5, 2023
  • #21

google/tree-math
New release?

Is there a roadmap for the next release? Thanks.
  • muffgaga
  • 1
  • Opened 
    on Dec 1, 2022
  • #17

google/tree-math
`tm.unwrap`: Error when `out_vectors` is list [documentation]

import jax.numpy as jnp import tree_math as tm def f(x, y): return x, y x = y = tm.Vector(jnp.array(0.)) tm.unwrap(f, out_vectors = (True, False))(x, y) # (tree_math.Vector(DeviceArray(0., dtype=float32, ...
  • deasmhumhna
  • 1
  • Opened 
    on Nov 10, 2022
  • #16

google/tree-math
Unexpected input type for array?

I m not sure if I m using this right, but here s what I have: from tree_math import Vector from typing import Any, Callable, TypeVar from jax.scipy.optimize import minimize T = TypeVar( T ) def tree_minimize(fun: ...
  • NeilGirdhar
  • 2
  • Opened 
    on Oct 17, 2022
  • #15

google/tree-math
import struct

How should I go about importing struct? Thanks! My attempt below fails -- import tree_math @tree_math.struct class Point: x: float y: float AttributeError: module tree_math has no attribute ...
  • pharringtonp19
  • 1
  • Opened 
    on Aug 10, 2022
  • #14

google/tree-math
How well does tree-math support computation on multiple devices?

Wondering how well tree-math supports computation on multiple devices? Let s say we have a pytree of tensors of different dimensions and want to perform some operations on each of them with tree-math, ...
  • connection-on-fiber-bundles
  • 1
  • Opened 
    on Apr 21, 2022
  • #10

google/tree-math
Transform for defining dataclasses with VectorMixin like flax.struct

It would be nice to have an easy way to define dataclasses that are also tree-math vectors. We could borrow the syntax of flax.struct here: https://flax.readthedocs.io/en/latest/flax.struct.html Example ...
  • shoyer
  • 2
  • Opened 
    on Feb 15, 2022
  • #9

google/tree-math
Operations should allow shape broadcasting

It seems like the current implementation doesn t allow broadcasting arguments. Here s an example for normalizing leafs. import tree_math as tm import jax import jax.numpy as jnp a = jnp.ones(10) b = ...
  • GeoffNN
  • 5
  • Opened 
    on Jan 3, 2022
  • #8

google/tree-math
Matrix support

I d like to add a Matrix class to complement Vector. A key design question is what this needs to support. In particular: do we need to support multiple axes that correspond to flattened pytrees, or is ...
  • shoyer
  • 11
  • Opened 
    on Dec 29, 2021
  • #6

google/tree-math
Rename wrap/unwrap?

These are rather undescriptive names, and I guess most people (including myself!) will have to guess check to keep them straight. :) Some other possibilities, in rough order of increasing descriptiveness: ...
  • shoyer
  • 2
  • Opened 
    on Dec 29, 2021
  • #5
Issue origami icon

Learn how you can use GitHub Issues to plan and track your work.

Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub Issues
ProTip! 
Press the
/
key to activate the search input again and adjust your query.
Issue origami icon

Learn how you can use GitHub Issues to plan and track your work.

Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub Issues
ProTip! 
Restrict your search to the title by using the in:title qualifier.
Issue search results · GitHub